Agents_Course_Final_Assignment

Runtime error

sanjeed5 commited on May 17

Commit

2683590

1 Parent(s): 903cbfd

Update agent.py to switch the LLM model from Gemini to OpenAI's GPT-4.1, adjusting the API key accordingly for improved performance and compatibility.

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -10,7 +10,7 @@ import io # Added for StringIO
 load_dotenv()
 # Configure the underlying LLM model for the agent - using a vision-capable model
-smol_model = LiteLLMModel(model_id="gemini/gemini-2.5-flash-preview-04-17", api_key=os.getenv("GEMINI_API_KEY"))
 # Instantiate the CodeAgent with authorized imports
 # Removed base64, litellm. Kept PIL for now, though framework might handle image display.

 load_dotenv()
 # Configure the underlying LLM model for the agent - using a vision-capable model
+smol_model = LiteLLMModel(model_id="openai/gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
 # Instantiate the CodeAgent with authorized imports
 # Removed base64, litellm. Kept PIL for now, though framework might handle image display.