Update model.py
Browse files
model.py
CHANGED
|
@@ -163,7 +163,8 @@ def build_graph(provider: str = "groq"):
|
|
| 163 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 164 |
elif provider == "groq":
|
| 165 |
# Groq https://console.groq.com/docs/models
|
| 166 |
-
|
|
|
|
| 167 |
elif provider == "huggingface":
|
| 168 |
# TODO: Add huggingface endpoint
|
| 169 |
llm = ChatHuggingFace(
|
|
|
|
| 163 |
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
|
| 164 |
elif provider == "groq":
|
| 165 |
# Groq https://console.groq.com/docs/models
|
| 166 |
+
api_key = os.getenv('GROQ_API')
|
| 167 |
+
llm = ChatGroq(model="qwen-qwq-32b", temperature=0,api_key = api_key) # optional : qwen-qwq-32b gemma2-9b-it
|
| 168 |
elif provider == "huggingface":
|
| 169 |
# TODO: Add huggingface endpoint
|
| 170 |
llm = ChatHuggingFace(
|