supratipb commited on
Commit
d1c1821
Β·
verified Β·
1 Parent(s): e7735b0

Upload 2 files

Browse files
Files changed (2) hide show
  1. .env +1 -0
  2. agent.py +2 -2
.env CHANGED
@@ -1,3 +1,4 @@
1
  GOOGLE_APPLICATION_CREDENTIALS=./ivory-bonus-284511-ece2419adbb2.json
2
  SUPABASE_URL=https://qgggbukkivjakulcmprh.supabase.co
3
  SUPABASE_SERVICE_KEY=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InFnZ2didWtraXZqYWt1bGNtcHJoIiwicm9sZSI6InNlcnZpY2Vfcm9sZSIsImlhdCI6MTc1MjIxOTE4NSwiZXhwIjoyMDY3Nzk1MTg1fQ.XxiYK6LzPGWKPiCBbF_moqfmpwCBsFLzw0aBNgxpN9M
 
 
1
  GOOGLE_APPLICATION_CREDENTIALS=./ivory-bonus-284511-ece2419adbb2.json
2
  SUPABASE_URL=https://qgggbukkivjakulcmprh.supabase.co
3
  SUPABASE_SERVICE_KEY=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InFnZ2didWtraXZqYWt1bGNtcHJoIiwicm9sZSI6InNlcnZpY2Vfcm9sZSIsImlhdCI6MTc1MjIxOTE4NSwiZXhwIjoyMDY3Nzk1MTg1fQ.XxiYK6LzPGWKPiCBbF_moqfmpwCBsFLzw0aBNgxpN9M
4
+ GROQ_API_KEY=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
agent.py CHANGED
@@ -153,7 +153,7 @@ tools = [
153
  ]
154
 
155
  # Build graph function
156
- def build_graph(provider: str = "google"):
157
  """Build the graph"""
158
  # Load environment variables from .env file
159
  if provider == "google":
@@ -161,7 +161,7 @@ def build_graph(provider: str = "google"):
161
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
162
  elif provider == "groq":
163
  # Groq https://console.groq.com/docs/models
164
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0, api_key=os.getenv("GROQ_API_KEY")) # optional : qwen-qwq-32b gemma2-9b-it
165
  elif provider == "huggingface":
166
  # TODO: Add huggingface endpoint
167
  llm = ChatHuggingFace(
 
153
  ]
154
 
155
  # Build graph function
156
+ def build_graph(provider: str = "groq"):
157
  """Build the graph"""
158
  # Load environment variables from .env file
159
  if provider == "google":
 
161
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
162
  elif provider == "groq":
163
  # Groq https://console.groq.com/docs/models
164
+ llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
165
  elif provider == "huggingface":
166
  # TODO: Add huggingface endpoint
167
  llm = ChatHuggingFace(