Upload folder using huggingface_hub

app/services/llm.py

@@ -4,6 +4,22 @@ from app.config import get_settings
 from app.models.question import Answer
 from app.models.checklist import ChecklistItem
 import json
+import re
+
+
+def extract_json(text: str) -> dict:
+    """Извлекает JSON из текста, даже если он обернут в markdown code blocks"""
+    # Пробуем найти JSON в code blocks
+    json_match = re.search(r'```(?:json)?\s*([\s\S]*?)```', text)
+    if json_match:
+        text = json_match.group(1).strip()
+    
+    # Пробуем найти JSON объект напрямую
+    json_match = re.search(r'\{[\s\S]*\}', text)
+    if json_match:
+        text = json_match.group(0)
+    
+    return json.loads(text)
 
 
 class LLMService:
@@ -42,7 +58,7 @@ class LLMService:
             ]
         )
         
-        result = json.loads(response.content[0].text)
+        result = extract_json(response.content[0].text)
         return result["questions"]
 
     def analyze_round_and_generate_questions(
@@ -99,7 +115,7 @@ class LLMService:
             ]
         )
         
-        return json.loads(response.content[0].text)
+        return extract_json(response.content[0].text)
 
     def generate_checklist(
         self,
@@ -162,7 +178,7 @@ class LLMService:
             ]
         )
         
-        return json.loads(response.content[0].text)
+        return extract_json(response.content[0].text)
 
 
 def get_llm_service() -> LLMService: