Spaces:

aleixlopezpascual
/

final_assignment_v3

Sleeping

aleixlopezpascual commited on May 24

Commit

7f5f422

1 Parent(s): 0c4a5aa

new gemini

Files changed (2) hide show

agent.py CHANGED Viewed

@@ -248,7 +248,7 @@ def build_graph(provider: str = "google"):
     # Load environment variables from .env file
     if provider == "google":
         # Google Gemini
-        llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     # elif provider == "groq":
     #     # Groq https://console.groq.com/docs/models
     #     llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it

     # Load environment variables from .env file
     if provider == "google":
         # Google Gemini
+        llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-preview-04-17", temperature=0)
     # elif provider == "groq":
     #     # Groq https://console.groq.com/docs/models
     #     llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it

test.ipynb CHANGED Viewed

The diff for this file is too large to render. See raw diff