antonioschiro commited on
Commit
14cafef
·
verified ·
1 Parent(s): 13074f3

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +8 -3
agent.py CHANGED
@@ -6,7 +6,10 @@ from langchain_groq import ChatGroq
6
  from langgraph.prebuilt import ToolNode
7
  from langgraph.graph import START, StateGraph, END
8
  from langgraph.prebuilt import tools_condition
9
- from tools import websearch
 
 
 
10
  from prompt import text_prompt
11
  from dotenv import load_dotenv
12
  #from PIL import Image
@@ -18,11 +21,13 @@ os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")
18
  class State(TypedDict):
19
  messages: Annotated[list[AnyMessage], add_messages]
20
 
21
- tools = [websearch]
 
 
22
 
23
  llm = ChatGroq(
24
  model="deepseek-r1-distill-llama-70b",
25
- temperature=0.1,
26
  max_tokens= None,
27
  reasoning_format="parsed",
28
  timeout=None,
 
6
  from langgraph.prebuilt import ToolNode
7
  from langgraph.graph import START, StateGraph, END
8
  from langgraph.prebuilt import tools_condition
9
+ from tools import (retriever, web_search, wiki_search, youtube_analysis,
10
+ add_numbers, subtract_numbers, multiply_numbers, divide_numbers, modulus_numbers,
11
+ detect_objects, run_python
12
+ )
13
  from prompt import text_prompt
14
  from dotenv import load_dotenv
15
  #from PIL import Image
 
21
  class State(TypedDict):
22
  messages: Annotated[list[AnyMessage], add_messages]
23
 
24
+ tools = [retriever, web_search, wiki_search, youtube_analysis,
25
+ add_numbers, subtract_numbers, multiply_numbers, divide_numbers, modulus_numbers,
26
+ detect_objects, run_python]
27
 
28
  llm = ChatGroq(
29
  model="deepseek-r1-distill-llama-70b",
30
+ temperature=0.0,
31
  max_tokens= None,
32
  reasoning_format="parsed",
33
  timeout=None,