model change

analyzer.py

@@ -1,38 +1,40 @@
-# Анализ одного портфеля
-import requests
-from fetch import extract_portfolio_id
+import httpx
 from prompts import ONE_PROMPT
+from fetch import extract_portfolio_id
 from openai import OpenAI
+import asyncio
+
+
+async def get_portfolio_metrics(portfolio_id: str) -> dict | None:
+    url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1"
+    try:
+        async with httpx.AsyncClient(timeout=10) as client:
+            response = await client.get(url)
+            data = response.json().get("data", {}).get("extended", {})
+            return {k: v for k, v in data.items() if isinstance(v, (int, float))}
+    except Exception as e:
+        print(f"[API Error]: {e}")
+        return None
 
-def analyze_portfolio_streaming(text: str, client:OpenAI):
+
+def analyze_portfolio_streaming(text: str, client: OpenAI):
     portfolio_id = extract_portfolio_id(text)
     if not portfolio_id:
         yield "❗ Укажите корректный portfolioId или ссылку."
         return
 
+    yield "⏳ Загружаю метрики портфеля..."
+
+    metrics = asyncio.run(get_portfolio_metrics(portfolio_id))
+    if not metrics:
+        yield "❗ Не удалось получить метрики портфеля."
+        return
+
+    metrics_text = ", ".join([f"{k}: {v}" for k, v in metrics.items()])
+    prompt = f"Вот метрики портфеля: {metrics_text}. Проанализируй их. Дай общий отчёт на русском языке."
+
     try:
-        # Получение JSON по API
-        url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1"
-        response = requests.get(url)
-        json_data = response.json()
-        extended = json_data.get("data", {}).get("extended", {})
-
-        if not extended:
-            yield "❗ Метрики не найдены в ответе API."
-            return
-
-        # Передаём все числовые метрики, без фильтрации
-        metrics = {k: v for k, v in extended.items() if isinstance(v, (int, float))}
-        if not metrics:
-            yield "❗ Нет числовых метрик для анализа."
-            return
-
-        metrics_text = ", ".join([f"{k}: {v}" for k, v in metrics.items()])
-        prompt = f"Вот метрики портфеля: {metrics_text}. Проанализируй их. Дай общий отчёт на русском языке"
-
-        # Запрос к модели
         response_llm = client.chat.completions.create(
-            # model="nasiruddin15/Mistral-dolphin-2.8-grok-instract-2-7B-slerp",
             model="meta-llama/Meta-Llama-3.1-8B-Instruct",
             messages=[
                 {"role": "system", "content": ONE_PROMPT},
@@ -49,4 +51,4 @@ def analyze_portfolio_streaming(text: str, client:OpenAI):
                 yield partial
 
     except Exception as e:
-        yield f"❌ Ошибка при обработке: {e}"
+        yield f"❌ Ошибка при генерации ответа: {e}"

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 from prompts import TRADELINK_CONTEXT
 from fetch import extract_portfolio_id
-from fetch import fetch_metrics
+from fetch import fetch_metrics_async
 from openai import OpenAI
 from analyzer import analyze_portfolio_streaming
 from comparer import compare_portfolio

comparer.py

@@ -1,6 +1,6 @@
 
 from fetch import extract_portfolio_id
-from fetch import fetch_metrics
+from fetch import fetch_metrics_async
 from prompts import COMPARE_PROMPT
 from openai import OpenAI
 
@@ -15,8 +15,8 @@ def compare_portfolio(text1: str, text2: str, client: OpenAI):
         yield "❗ Один или оба portfolioId некорректны."
         return
 
-    m1 = fetch_metrics(id1)
-    m2 = fetch_metrics(id2)
+    m1 = fetch_metrics_async(id1)
+    m2 = fetch_metrics_async(id2)
     if not m1 or not m2:
         yield "❗ Не удалось получить метрики одного из портфелей."
         return

fetch.py

@@ -1,5 +1,6 @@
 import re
 import requests
+import httpx
 
 #Извлечение UUID из строки
 def extract_portfolio_id(text: str) -> str | None:
@@ -9,28 +10,36 @@ def extract_portfolio_id(text: str) -> str | None:
     )
     return match.group(0) if match else None
 
-def fetch_metrics(portfolio_id: str) -> dict | None:
+async def fetch_metrics_async(portfolio_id: str) -> dict | None:
+    url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1"
     try:
-        url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1"
-        response = requests.get(url)
-        response.raise_for_status()
-        extended = response.json().get("data", {}).get("extended", {})
-        keys = [
-            # 🔹 Доходность и риск
-            "alphaRatio", "betaRatio", "cagr", "sharpe", "sortino", "volatility",
-            "kSortino", "kCalmar", "kSharpe",
-
-            # 🔹 Просадки
-            "maxDD", "mddDuration", "maxBalance",
-
-            # 🔹 Поведение стратегии
-            "losingDays", "winningDays", "selfProfitRate",
+        async with httpx.AsyncClient(timeout=10) as client:
+            resp = await client.get(url)
+            extended = resp.json().get("data", {}).get("extended", {})
+            return {k: v for k, v in extended.items() if isinstance(v, (int, float))}
+    except Exception as e:
+        print(f"[API ошибка]: {e}")
+        return None
 
-            # 🔹 Актуальные показатели доходности
-            "lastWeekNetProfit", "lastMonthNetProfit", "lastQuarterGrowth", "lastYearNetGrowth"
-        ]
+        # print(f"[API ошибка]: {e}")
+        # return None
+        #
+        # keys = [
+        #     # 🔹 Доходность и риск
+        #     "alphaRatio", "betaRatio", "cagr", "sharpe", "sortino", "volatility",
+        #     "kSortino", "kCalmar", "kSharpe",
+        #
+        #     # 🔹 Просадки
+        #     "maxDD", "mddDuration", "maxBalance",
+        #
+        #     # 🔹 Поведение стратегии
+        #     "losingDays", "winningDays", "selfProfitRate",
+        #
+        #     # 🔹 Актуальные показатели доходности
+        #     "lastWeekNetProfit", "lastMonthNetProfit", "lastQuarterGrowth", "lastYearNetGrowth"
+        # ]
 
-        return {k: extended[k] for k in keys if isinstance(extended.get(k), (int, float))}
-    except:
-        return None
+    #     return {k: extended[k] for k in keys if isinstance(extended.get(k), (int, float))}
+    # except:
+    #     return None
 

requirements.txt

@@ -1,3 +1,4 @@
 gradio>=4.29.0
 openai>=1.30.1
 requests
+httpx