Spaces:
Sleeping
Sleeping
File size: 1,801 Bytes
1441b40 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from groq import Groq
import json
import sys
import os
# Add parent directory to path for imports
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from config.settings import settings
from utils.prompts import SYSTEM_PROMPT, create_user_prompt
class GroqService:
def __init__(self):
self.client = Groq(api_key=settings.GROQ_API_KEY)
self.model_id = settings.MODEL_ID
async def analyze_provider_notes(self, provider_notes: str) -> dict:
"""
Analyze provider notes and return ICD and CPT codes with explanations
"""
try:
# Create the chat completion with system and user prompts
chat_completion = self.client.chat.completions.create(
messages=[
{
"role": "system",
"content": SYSTEM_PROMPT
},
{
"role": "user",
"content": create_user_prompt(provider_notes)
}
],
model=self.model_id,
temperature=0.1, # Low temperature for consistent, factual outputs
response_format={"type": "json_object"} # Force JSON response
)
# Extract and parse the response
response_content = chat_completion.choices[0].message.content
parsed_response = json.loads(response_content)
return parsed_response
except json.JSONDecodeError as e:
raise ValueError(f"Failed to parse JSON response from model: {str(e)}")
except Exception as e:
raise Exception(f"Error calling Groq API: {str(e)}")
groq_service = GroqService() |