bitcloud2 commited on
Commit
d35ad8f
·
1 Parent(s): d7a6b7d

cleaning final answer

Browse files
Files changed (2) hide show
  1. Changelog.md +1 -0
  2. gemini_model.py +5 -0
Changelog.md CHANGED
@@ -12,6 +12,7 @@
12
  - Simplified error handling in `run_and_submit_all` by removing redundant try-except block around `instantiate_agent` call.
13
  - Updated `run_and_submit_all` to call `fetch_questions` and handle its return values.
14
  - Updated `app_dev.py` to import and use `SmolAgent` from `my_agent.py`.
 
15
 
16
  ### Deprecated
17
 
 
12
  - Simplified error handling in `run_and_submit_all` by removing redundant try-except block around `instantiate_agent` call.
13
  - Updated `run_and_submit_all` to call `fetch_questions` and handle its return values.
14
  - Updated `app_dev.py` to import and use `SmolAgent` from `my_agent.py`.
15
+ - Added logic to `GeminiApiModel` to extract text after "FINAL ANSWER: " marker in responses.
16
 
17
  ### Deprecated
18
 
gemini_model.py CHANGED
@@ -125,6 +125,11 @@ class GeminiApiModel(ApiModel):
125
  part = response_json["candidates"][0]["content"]["parts"][0]
126
  if "text" in part:
127
  content = part["text"]
 
 
 
 
 
128
  # Simulate token counts if available, otherwise default to 0
129
  # The basic generateContent API might not return usage directly in the main response
130
  # It might be in safetyRatings or other metadata if enabled/available.
 
125
  part = response_json["candidates"][0]["content"]["parts"][0]
126
  if "text" in part:
127
  content = part["text"]
128
+ # Check for "FINAL ANSWER: " and extract the rest of the string
129
+ final_answer_marker = "FINAL ANSWER: "
130
+ if final_answer_marker in content:
131
+ content = content.split(final_answer_marker)[-1].strip()
132
+
133
  # Simulate token counts if available, otherwise default to 0
134
  # The basic generateContent API might not return usage directly in the main response
135
  # It might be in safetyRatings or other metadata if enabled/available.