amkyawdev commited on
Commit
8b37747
·
verified ·
1 Parent(s): 24a95e0

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
- result = await generate_response(messages)
 
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
- model = model or settings.GROQ_MODEL
 
 
 
41
 
42
  response = groq_client.chat.completions.create(
43
- model=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
- model = model or settings.GROQ_MODEL
65
 
66
  stream = groq_client.chat.completions.create(
67
- model=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,