Commit
·
237dd50
1
Parent(s):
999f7f9
remove groq
Browse files
agent.py
CHANGED
@@ -152,15 +152,15 @@ tools = [
|
|
152 |
]
|
153 |
|
154 |
# Build graph function
|
155 |
-
def build_graph(provider: str = "
|
156 |
"""Build the graph"""
|
157 |
# Load environment variables from .env file
|
158 |
if provider == "google":
|
159 |
# Google Gemini
|
160 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
161 |
-
elif provider == "groq":
|
162 |
-
|
163 |
-
|
164 |
elif provider == "huggingface":
|
165 |
# TODO: Add huggingface endpoint
|
166 |
llm = ChatHuggingFace(
|
|
|
152 |
]
|
153 |
|
154 |
# Build graph function
|
155 |
+
def build_graph(provider: str = "google"):
|
156 |
"""Build the graph"""
|
157 |
# Load environment variables from .env file
|
158 |
if provider == "google":
|
159 |
# Google Gemini
|
160 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
161 |
+
# elif provider == "groq":
|
162 |
+
# # Groq https://console.groq.com/docs/models
|
163 |
+
# llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
164 |
elif provider == "huggingface":
|
165 |
# TODO: Add huggingface endpoint
|
166 |
llm = ChatHuggingFace(
|