mirror of
http://88.130.71.182:3000/BlitTech/contexta_be.git
synced 2026-06-12 23:23:21 +00:00
fixed the RAg in test pipeline issue
This commit is contained in:
@@ -31,14 +31,9 @@ class LLMService:
|
||||
return await self._call_openai(messages, model, max_tokens, temperature)
|
||||
except Exception as e:
|
||||
logger.error(f"LLM error ({provider}/{model}): {e}")
|
||||
# Fallback to a basic model if available
|
||||
if model != "accounts/fireworks/models/kimi-k2-instruct-0905" and settings.fireworks_api_key:
|
||||
return await self._call_fireworks(
|
||||
messages,
|
||||
"accounts/fireworks/models/kimi-k2-instruct-0905",
|
||||
max_tokens,
|
||||
temperature,
|
||||
)
|
||||
fallback = "accounts/fireworks/models/llama-v3p3-70b-instruct"
|
||||
if model != fallback and settings.fireworks_api_key:
|
||||
return await self._call_fireworks(messages, fallback, max_tokens, temperature)
|
||||
raise
|
||||
|
||||
async def _call_fireworks(
|
||||
|
||||
Reference in New Issue
Block a user