Modification

src/agent/agenttools.py

@@ -49,6 +49,17 @@ class ChatBotFunctionTools:
         prompt = f"Explain the relationship between these concepts: {', '.join(concepts)}"
         return self.generator.complete(prompt)
 
+    @retry(max_retries=5, delay=1)
+    def code_generator(self, query):
+        logging.info(f"Tool Call: copilot('{query}')")
+        prompt = (
+            "You are a Python Expert.\n"
+            "Give the python source code as asked like copilot or help to debug a particular code block:\n"
+            f"{query}\n"
+            "Keep it compact and dont give much theory. Explain code blocks only."
+        )
+        return self.generator.complete(prompt)
+
     @retry(max_retries=5, delay=1)
     def llm_query(self, concepts):
         logging.info(f"Tool Call: llm_query('{concepts}')")
@@ -63,7 +74,8 @@ class ChatBotFunctionTools:
             "algo_visualizer": FunctionTool.from_defaults(fn=self.visualize_algorithm),
             "concept_combiner": FunctionTool.from_defaults(fn=self.concept_combiner),
             "math_concept": FunctionTool.from_defaults(fn=self.math_concept),
-            "llm_query": FunctionTool.from_defaults(fn=self.llm_query)
+            "llm_query": FunctionTool.from_defaults(fn=self.llm_query),
+            "code_generator": FunctionTool.from_defaults(fn=self.code_generator)
         }
 
 

src/agent/custom_agent.py

@@ -36,7 +36,8 @@ class LearningAgent:
         prompt = f"""Analyze this learning query and select appropriate tools also form the condensed query based on
                     previous_questions and Query:
                     For Choosing tool properly analyze the condensed query and then decide. Choose multiple if its necessary based on condensed query
-                    Its mandatory to select to atleast 1 tool
+                    Its mandatory to select to atleast 1 tool.
+                    If asked for any query related to code generation dont include theory rather give code and its explaination
                     Query: {query}
                     Previous Query: {previous_questions}
                     Available Tools: {list(self.tools.keys())}