Skip to content

Commit 193ab3c

Browse files
committed
Update eval_frames_benchmark.py
add more timeout as fetching and going over wikipedia pages can take time.
1 parent 2d3b951 commit 193ab3c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/eval_frames_benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def generate_llm_prompt(prompt: str, wiki_links: List[str]) -> str:
3434
return f"Here are the relevant Wikipedia articles:\n{wiki_links}\n\nBased on all the information, answer the query. \n\nQuery: {prompt}\n\n"
3535

3636
def get_llm_response(prompt: str, model: str) -> str:
37-
response = client.chat.completions.create(
37+
response = client.with_options(timeout=1000.0).chat.completions.create(
3838
model=model,
3939
messages=[
4040
{"role": "system", "content": "You are a helpful assistant."},

0 commit comments

Comments
 (0)