Remove prompt over-engineering

Author: hrshdhgd

src/llm_change_agent/llm_agent.py

@@ -12,7 +12,6 @@
     OPENAI_PROVIDER,
 )
 from llm_change_agent.utils.llm_utils import (
-    augment_prompt,
     execute_agent,
     get_anthropic_models,
     get_default_model_for_provider,
@@ -99,6 +98,6 @@ def run(self):
         """Run the LLM Change Agent."""
         llm_config = self._get_llm_config()
         self.llm = llm_factory(llm_config)
-        response = execute_agent(llm=self.llm, prompt=augment_prompt(self.prompt), external_rag_docs=self.docs)
+        response = execute_agent(llm=self.llm, prompt=self.prompt, external_rag_docs=self.docs)
         pprint(response["output"])
         return response["output"]

src/llm_change_agent/utils/llm_utils.py

@@ -31,8 +31,6 @@
     KGCL_SCHEMA,
     OLLAMA_PROVIDER,
     ONTOLOGIES_AS_DOC_MAP,
-    ONTOLOGIES_URL,
-    # ONTODIFF_DOCS,
     OPENAI_KEY,
     OPENAI_PROVIDER,
     VECTOR_DB_NAME,
@@ -339,30 +337,23 @@ def _chunk_list(lst, chunk_size):
     tools = [tool, compress_iri]
     template = get_issue_analyzer_template()
     react_agent = create_react_agent(llm=llm, tools=tools, prompt=template)
-    agent_executor = AgentExecutor(agent=react_agent, tools=tools, handle_parsing_errors=True, verbose=True)
+    agent_executor = AgentExecutor(
+        agent=react_agent,
+        tools=tools,
+        handle_parsing_errors=True,
+        verbose=True,
+        return_intermediate_steps=True,
+    )
     logger.info("Agent executor created successfully.")
-
     return agent_executor.invoke(
         {
             "input": prompt,
             "grammar": grammar["lark"],
             "explanation": grammar["explanation"],
-            "ontology_urls": ONTOLOGIES_URL,
         }
     )
 
 
-def augment_prompt(prompt: str):
-    """Augment the prompt with additional information."""
-    return f"""
-        Give me all relevant KGCL commands based on this request: \n\n
-        + {prompt} +
-        \n\n
-        Return as a python list object which will be passed to another tool.
-        Each element of the list should be enlosed in double quotes.
-        """
-
-
 def extract_commands(command):
     """Extract the command from the list."""
     # Remove markdown markers