mjmuskan03/coding-agent

1from crewai import Agent, Task
2from rag_coding import RagCodingAgent
3from dotenv import load_dotenv
4import os
5
6load_dotenv()
7api_key = os.getenv("OPENAI_API_KEY")
8
9class CodeMentorCrew:
10    def __init__(self):
11        self.rag_agent = RagCodingAgent()
12        self.coding_agent = Agent(
13            role="Code Mentor",
14            goal="Assist developers by generating accurate and efficient code solutions",
15            backstory="You are an expert coder with years of experience in Python, JavaScript, and more, dedicated to helping developers solve coding problems.",
16            tools=[self.rag_agent.process],
17            verbose=True
18        )
19
20    def create_task(self, query):
21        task = Task(
22            description=f"Generate code for the following query: {query}",
23            agent=self.coding_agent,
24            expected_output="A complete and working code snippet"
25        )
26        return task
27
28    def run(self, query):
29        task = self.create_task(query)
30        return task.execute()
31
32if __name__ == "__main__":
33    crew = CodeMentorCrew()
34    query = "Write a Python function to sort a list"
35    result = crew.run(query)
36    print("Code Solution:\n", result)