chaowenguo commited on
Commit
4f826c0
·
verified ·
1 Parent(s): 232e1ff

Update app/services/llm.py

Browse files
Files changed (1) hide show
  1. 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 = config.app.get("cloudflare_model_name")
70
- account_id = config.app.get("cloudflare_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/11b1d730ea208e2e78adc22e1e2e8059/ai/run/@cf/meta/llama-4-scout-17b-16e-instruct",
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": [