few changes - 2

Shubhamsaboo · Dec 12, 2024 · ece3f8e · ece3f8e
1 parent be85460
commit ece3f8e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/ai_agent_tutorials/ai_legal_agent_team/local_ai_legal_agent_team/local_legal_agent.py b/ai_agent_tutorials/ai_legal_agent_team/local_ai_legal_agent_team/local_legal_agent.py
@@ -84,7 +84,7 @@ def main():
                 legal_researcher = Agent(
                     name="Legal Researcher",
                     role="Legal research specialist",
-                    model=Ollama(id="llama3.1"),  
+                    model=Ollama(id="llama3.1:8b"),  
                     knowledge=st.session_state.knowledge_base,
                     search_knowledge=True,
                     instructions=[
@@ -98,7 +98,7 @@ def main():
                 contract_analyst = Agent(
                     name="Contract Analyst",
                     role="Contract analysis specialist",
-                    model=Ollama(id="llama3.1"),
+                    model=Ollama(id="llama3.1:8b"),
                     knowledge=knowledge_base,
                     search_knowledge=True,
                     instructions=[
@@ -112,7 +112,7 @@ def main():
                 legal_strategist = Agent(
                     name="Legal Strategist", 
                     role="Legal strategy specialist",
-                    model=Ollama(id="llama3.1"),
+                    model=Ollama(id="llama3.1:8b"),
                     knowledge=knowledge_base,
                     search_knowledge=True,
                     instructions=[

diff --git a/ai_agent_tutorials/ai_legal_agent_team/local_ai_legal_agent_team/requirements.txt b/ai_agent_tutorials/ai_legal_agent_team/local_ai_legal_agent_team/requirements.txt
@@ -1,6 +1,4 @@
-phidata==2.5.33
+phidata==2.6.7
 streamlit==1.40.2
 qdrant-client==1.12.1
-pypdf
-python-dotenv
-ollama==0.1.6
+ollama==0.4.4