File size: 2,943 Bytes
bd702b9 e225216 bd702b9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
answer = response.choices[0].message.content.strip()
final_line = ""
for line in answer.splitlines():
if line.strip().lower().startswith("final answer:"):
final_line = line.split(":", 1)[-1].strip(" .\"'")
break
bads = [
"", "unknown", "unable to determine", "unable to provide page numbers",
"unable to access video content directly", "unable to analyze video content",
"unable to determine without code", "unable to determine without file",
"follow the steps to locate the paper and find the nasa award number in the acknowledgment section",
"i am unable to view images or access external content directly", "unable to determine without access to the file",
"no results found", "n/a", "[your final answer]", "i'm sorry", "i apologize"
]
if final_line.lower() in bads or final_line.lower().startswith("unable") or final_line.lower().startswith("i'm sorry") or final_line.lower().startswith("i apologize"):
# --- Try to extract a plausible answer from web or file ---
# Example: For numbers
numbers = re.findall(r'\b\d{2,}\b', search_snippet)
if numbers:
return numbers[0]
# Example: For possible names (capitalize words)
words = re.findall(r'\b[A-Z][a-z]{2,}\b', search_snippet)
if words:
return words[0]
# Example: For Excel, code, or file extraction, return the result
if file_result:
file_numbers = re.findall(r'\b\d{2,}\b', file_result)
if file_numbers:
return file_numbers[0]
file_words = re.findall(r'\b[A-Z][a-z]{2,}\b', file_result)
if file_words:
return file_words[0]
# --- Try to re-ask the LLM to answer "without apologies" ---
retry_prompt = (
"Based ONLY on the search results and/or file content above, return a direct answer to the question. "
"If you do not know, make your best plausible guess. Do NOT apologize or say you cannot assist. "
f"File: {file_result}\n\nWeb: {search_snippet}\n\nQuestion: {question}\nFINAL ANSWER:"
)
response2 = self.llm.chat.completions.create(
model="gpt-4o",
messages=[{"role": "system", "content": retry_prompt}],
temperature=0.1,
max_tokens=128,
)
retry_answer = response2.choices[0].message.content.strip()
for line in retry_answer.splitlines():
if line.strip().lower().startswith("final answer:"):
return line.split(":", 1)[-1].strip(" .\"'")
if retry_answer:
return retry_answer.strip(" .\"'")
return final_line |