openfree commited on
Commit
3564514
·
verified ·
1 Parent(s): bad7e32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -97,7 +97,7 @@ def brave_search(query: str, count: int = 5) -> List[Dict]:
97
  except Exception as e:
98
  return [{"title": "Error", "url": "", "snippet": str(e)}]
99
 
100
- def call_llm(messages: List[Dict], temperature: float = 0.6, max_tokens: int = 1024) -> str:
101
  """Call Fireworks AI API"""
102
  api_key = get_secret("FIREWORKS_API_KEY", "")
103
  if not api_key:
@@ -319,7 +319,7 @@ Please provide a detailed, well-structured response that:
319
  {"role": "user", "content": user_msg}
320
  ]
321
 
322
- return call_llm(messages, temperature=0.4, max_tokens=1500)
323
 
324
  # --------------- Streamlit UI ---------------
325
 
 
97
  except Exception as e:
98
  return [{"title": "Error", "url": "", "snippet": str(e)}]
99
 
100
+ def call_llm(messages: List[Dict], temperature: float = 0.6, max_tokens: int = 4000) -> str:
101
  """Call Fireworks AI API"""
102
  api_key = get_secret("FIREWORKS_API_KEY", "")
103
  if not api_key:
 
319
  {"role": "user", "content": user_msg}
320
  ]
321
 
322
+ return call_llm(messages, temperature=0.4, max_tokens=4000)
323
 
324
  # --------------- Streamlit UI ---------------
325