wip: multi-agents

multi-agents-analysis/README.md

@@ -0,0 +1,34 @@
+# LLM Multi Agent Race Analysis System
+
+This project utilizes a multi-agent system to analyze race data from the FastF1 library using CrewAI and LangChain. The system is designed to provide detailed insights and comparisons between drivers in a race session.
+
+## Setup Instructions
+
+1. **Create and activate a Conda virtual environment:**
+
+   ```sh
+   conda create --name race-analysis python=3.11
+   conda activate race-analysis
+   ```
+
+2. **Install the dependencies:**
+
+   ```sh
+   conda env create --file environment.yml
+   ```
+
+3. **Paste your OpenAI key in the .env file.**
+
+4. **Provide basic context for the analysis system:**
+
+   - Create a file named `context.md` in the root directory.
+   - Add a brief text in `context.md` to provide basic context to the analysis system.
+
+5. **Run the main script to start the race analysis:**
+   ```sh
+   python main.py
+   ```
+
+## Usage
+
+To run the analysis, execute the following command:

multi-agents-analysis/agents.py

@@ -1,4 +1,7 @@
 from crewai import Agent
+from crewai_tools import (
+    PGSearchTool,
+)
 from textwrap import dedent
 from langchain.llms import OpenAI, Ollama
 from langchain_openai import ChatOpenAI
@@ -9,6 +12,8 @@ from langchain_openai import ChatOpenAI
 # You can also define custom tasks in tasks.py
 class CustomAgents:
     def __init__(self) -> None:
+        self.db_search_tool = PGSearchTool(
+            db_uri='sqlite:///laps.db', table_name='laps')
         # self.OpenAIGPT35 = ChatOpenAI(model='gpt-3.5-turbo', temperature=0.7)
         self.Ollama = Ollama(
             model="internlm2", base_url="http://localhost:11434", temperature=0.1)
@@ -29,7 +34,8 @@ class CustomAgents:
                         between them."""),
             allow_delegation=False,
             verbose=True,
-            llm=self.Ollama
+            llm=self.Ollama,
+            tools=[self.db_search_tool]
         )
 
     def race_engineer(self):

multi-agents-analysis/main.py

@@ -43,13 +43,13 @@ class CustomCrew:
         agents = CustomAgents()
         tasks = CustomTasks()
 
-        custom_agent_1 = agents.data_analyst()
+        data_analyst_agent = agents.data_analyst()
         custom_agent_2 = agents.race_engineer()
 
-        custom_task_1 = tasks.task_1_name(custom_agent_1, 'foo', 'var')
+        custom_task_1 = tasks.task_1_name(data_analyst_agent, 'foo', 'var')
         custom_task_2 = tasks.task_2_name(custom_agent_2)
 
-        crew = Crew(agents=[custom_agent_1, custom_agent_2], tasks=[
+        crew = Crew(agents=[data_analyst_agent, custom_agent_2], tasks=[
                     custom_task_1, custom_task_2], verbose=True)
 
         result = crew.kickoff()