Spaces:

Amodit
/

jan-contract

Sleeping

Amodit commited on Jan 31

Commit

188eb69

1 Parent(s): bd6f8a7

Force use of gemini-2.5-flash

Files changed (1) hide show

agents/general_assistant_agent.py CHANGED Viewed

@@ -10,7 +10,7 @@ try:
     api_key = os.getenv("GOOGLE_API_KEY")
     if api_key:
         client = genai.Client(api_key=api_key)
-        model_name = "gemini-2.0-flash-exp"
     else:
         print("CRITICAL WARNING: GOOGLE_API_KEY missing in general_assistant_agent.")
         client = None

     api_key = os.getenv("GOOGLE_API_KEY")
     if api_key:
         client = genai.Client(api_key=api_key)
+        model_name = "gemini-2.5-flash"
     else:
         print("CRITICAL WARNING: GOOGLE_API_KEY missing in general_assistant_agent.")
         client = None