Update app.py
Browse files
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 =
|
| 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=
|
| 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 |
|