Spaces:
Sleeping
Sleeping
HARDFIX: Force llama-3.1-8b-instant model everywhere - 70B deprecated
Browse files
src/api/agents/skill/orchestrator.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
"""Skill Agent Orchestrator"""
|
| 2 |
from typing import Dict, Any, List, Optional
|
| 3 |
-
from src.services.groq_service import generate_response
|
| 4 |
from src.api.core.logger import logger
|
| 5 |
|
| 6 |
|
|
@@ -32,7 +32,8 @@ class SkillOrchestrator:
|
|
| 32 |
]
|
| 33 |
|
| 34 |
try:
|
| 35 |
-
|
|
|
|
| 36 |
return {"skill_type": skill_type, "result": result, "status": "success"}
|
| 37 |
except Exception as e:
|
| 38 |
logger.error(f"Skill execution failed: {e}")
|
|
|
|
| 1 |
"""Skill Agent Orchestrator"""
|
| 2 |
from typing import Dict, Any, List, Optional
|
| 3 |
+
from src.services.groq_service import generate_response, DEFAULT_MODEL
|
| 4 |
from src.api.core.logger import logger
|
| 5 |
|
| 6 |
|
|
|
|
| 32 |
]
|
| 33 |
|
| 34 |
try:
|
| 35 |
+
# Use DEFAULT_MODEL directly
|
| 36 |
+
result = await generate_response(messages, model=DEFAULT_MODEL)
|
| 37 |
return {"skill_type": skill_type, "result": result, "status": "success"}
|
| 38 |
except Exception as e:
|
| 39 |
logger.error(f"Skill execution failed: {e}")
|
src/services/groq_service.py
CHANGED
|
@@ -8,12 +8,15 @@ from src.api.core.logger import logger
|
|
| 8 |
|
| 9 |
client: Optional[Groq] = None
|
| 10 |
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
def get_groq_client() -> Groq:
|
| 13 |
"""Get or create Groq client instance"""
|
| 14 |
global client
|
| 15 |
if client is None:
|
| 16 |
-
client = Groq(api_key=settings.GROQ_API_KEY)
|
| 17 |
return client
|
| 18 |
|
| 19 |
|
|
@@ -25,22 +28,16 @@ async def generate_response(
|
|
| 25 |
) -> str:
|
| 26 |
"""
|
| 27 |
Generate response using Groq API
|
| 28 |
-
|
| 29 |
-
Args:
|
| 30 |
-
messages: List of message dicts with 'role' and 'content'
|
| 31 |
-
model: Model name (defaults to settings.GROQ_MODEL)
|
| 32 |
-
temperature: Sampling temperature
|
| 33 |
-
max_tokens: Maximum tokens to generate
|
| 34 |
-
|
| 35 |
-
Returns:
|
| 36 |
-
Generated response string
|
| 37 |
"""
|
| 38 |
try:
|
| 39 |
groq_client = get_groq_client()
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
response = groq_client.chat.completions.create(
|
| 43 |
-
model=
|
| 44 |
messages=messages,
|
| 45 |
temperature=temperature,
|
| 46 |
max_tokens=max_tokens,
|
|
@@ -61,10 +58,10 @@ async def stream_response(
|
|
| 61 |
"""Stream response from Groq API"""
|
| 62 |
try:
|
| 63 |
groq_client = get_groq_client()
|
| 64 |
-
|
| 65 |
|
| 66 |
stream = groq_client.chat.completions.create(
|
| 67 |
-
model=
|
| 68 |
messages=messages,
|
| 69 |
temperature=temperature,
|
| 70 |
stream=True,
|
|
|
|
| 8 |
|
| 9 |
client: Optional[Groq] = None
|
| 10 |
|
| 11 |
+
# Default model - MUST use 8B as 70B is deprecated
|
| 12 |
+
DEFAULT_MODEL = "llama-3.1-8b-instant"
|
| 13 |
+
|
| 14 |
|
| 15 |
def get_groq_client() -> Groq:
|
| 16 |
"""Get or create Groq client instance"""
|
| 17 |
global client
|
| 18 |
if client is None:
|
| 19 |
+
client = Groq(api_key=settings.GROQ_API_KEY or "dummy")
|
| 20 |
return client
|
| 21 |
|
| 22 |
|
|
|
|
| 28 |
) -> str:
|
| 29 |
"""
|
| 30 |
Generate response using Groq API
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
"""
|
| 32 |
try:
|
| 33 |
groq_client = get_groq_client()
|
| 34 |
+
# Use DEFAULT_MODEL - 70b is deprecated!
|
| 35 |
+
model_name = model or getattr(settings, 'GROQ_MODEL', DEFAULT_MODEL) or DEFAULT_MODEL
|
| 36 |
+
|
| 37 |
+
print(f"[DEBUG] Using model: {model_name}")
|
| 38 |
|
| 39 |
response = groq_client.chat.completions.create(
|
| 40 |
+
model=model_name,
|
| 41 |
messages=messages,
|
| 42 |
temperature=temperature,
|
| 43 |
max_tokens=max_tokens,
|
|
|
|
| 58 |
"""Stream response from Groq API"""
|
| 59 |
try:
|
| 60 |
groq_client = get_groq_client()
|
| 61 |
+
model_name = model or getattr(settings, 'GROQ_MODEL', DEFAULT_MODEL) or DEFAULT_MODEL
|
| 62 |
|
| 63 |
stream = groq_client.chat.completions.create(
|
| 64 |
+
model=model_name,
|
| 65 |
messages=messages,
|
| 66 |
temperature=temperature,
|
| 67 |
stream=True,
|