Spaces:

real-jiakai
/

Agent_Course_Final_Assignment_Demo

Paused

real-jiakai commited on 4 days ago

Commit

ce2d7d4

verified ·

1 Parent(s): 97d8b63

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -72,7 +72,7 @@ class GeminiModel:
         resp = self.client.models.generate_content(
             model=self.model_name,
             contents=prompt,
-            generation_config=gtypes.GenerateContentConfig(
                 temperature=self.temperature,
                 max_output_tokens=self.max_tokens,
             ),
@@ -89,7 +89,7 @@ class GeminiModel:
         resp = self.client.models.generate_content(
             model=self.model_name,
             contents=contents,
-            generation_config=gtypes.GenerateContentConfig(
                 temperature=self.temperature,
                 max_output_tokens=self.max_tokens,
             ),

         resp = self.client.models.generate_content(
             model=self.model_name,
             contents=prompt,
+            config=gtypes.GenerateContentConfig(
                 temperature=self.temperature,
                 max_output_tokens=self.max_tokens,
             ),
         resp = self.client.models.generate_content(
             model=self.model_name,
             contents=contents,
+            config=gtypes.GenerateContentConfig(
                 temperature=self.temperature,
                 max_output_tokens=self.max_tokens,
             ),