Spaces:
Running
Running
Update app/services/llm.py
Browse files- app/services/llm.py +3 -3
app/services/llm.py
CHANGED
@@ -66,8 +66,8 @@ def _generate_response(prompt: str) -> str:
|
|
66 |
base_url = "***"
|
67 |
elif llm_provider == "cloudflare":
|
68 |
api_key = os.getenv('cloudflare')
|
69 |
-
model_name =
|
70 |
-
account_id =
|
71 |
base_url = "***"
|
72 |
elif llm_provider == "deepseek":
|
73 |
api_key = config.app.get("deepseek_api_key")
|
@@ -213,7 +213,7 @@ def _generate_response(prompt: str) -> str:
|
|
213 |
|
214 |
if llm_provider == "cloudflare":
|
215 |
response = requests.post(
|
216 |
-
f"https://api.cloudflare.com/client/v4/accounts/
|
217 |
headers={"Authorization": f'Bearer {api_key}'},
|
218 |
json={
|
219 |
"messages": [
|
|
|
66 |
base_url = "***"
|
67 |
elif llm_provider == "cloudflare":
|
68 |
api_key = os.getenv('cloudflare')
|
69 |
+
model_name = '@cf/meta/llama-4-scout-17b-16e-instruct'
|
70 |
+
account_id = '11b1d730ea208e2e78adc22e1e2e8059'
|
71 |
base_url = "***"
|
72 |
elif llm_provider == "deepseek":
|
73 |
api_key = config.app.get("deepseek_api_key")
|
|
|
213 |
|
214 |
if llm_provider == "cloudflare":
|
215 |
response = requests.post(
|
216 |
+
f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/{model_name}",
|
217 |
headers={"Authorization": f'Bearer {api_key}'},
|
218 |
json={
|
219 |
"messages": [
|