Glevel/New-planet-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Изменения:

Browse Source 
-добавлены нетворки в докер композ
-исправлен рутинг (баг пайчарма)
-запросы к ии агентам не проходят из-за ссл сертификата (пробовали отключить, но пока не выходит, нужно доделать)
This commit is contained in:
Артур Кузахметов
2025-12-19 00:57:13 +03:00
parent 24f4ce118f
commit a649fb1192
16 changed files with 433 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
new-planet-backend/app/api/deps.py
View File

@@ -7,7 +7,7 @@ from app.crud import user as crud_user
from app.services.auth_service import auth_service
from app.models.user import User
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="api/v1/auth/login")
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/v1/auth/login")
async def get_current_user(

22
new-planet-backend/app/api/v1/ai.py
View File

@@ -3,9 +3,27 @@ from sqlalchemy.ext.asyncio import AsyncSession
from app.db.session import get_db
from app.api.deps import get_current_active_user
from app.models.user import User
from app.schemas.ai import ChatRequest, ChatResponse, ScheduleGenerateRequest, ScheduleGenerateResponse
from app.models.ai_conversation import AIConversation
from app.schemas.ai import (
ChatRequest,
ChatResponse,
ScheduleGenerateRequest,
ScheduleGenerateResponse,
ConversationHistory,
ConversationListItem,
ScheduleUpdateRequest,
ScheduleUpdateResponse,
RecommendationRequest,
RecommendationResponse,
)
from app.services.chat_service import chat_service
from app.services.schedule_generator import schedule_generator
from app.services.cache_service import cache_service
from app.services.gigachat_service import gigachat_service
from app.crud import schedule as crud_schedule, task as crud_task
from app.schemas.task import TaskCreate
from app.schemas.schedule import ScheduleUpdate
from app.core.config import settings
router = APIRouter()
@@ -30,7 +48,6 @@ async def chat_with_ai(
detail=f"Chat error: {str(e)}"
)
@router.post("/schedule/generate", response_model=ScheduleGenerateResponse)
async def generate_schedule_ai(
request: ScheduleGenerateRequest,
@@ -58,3 +75,4 @@ async def generate_schedule_ai(
detail=f"Failed to generate schedule: {str(e)}"
)

16
new-planet-backend/app/core/config.py
View File

@@ -10,15 +10,15 @@ class Settings(BaseSettings):
DEBUG: bool = False
# Security
SECRET_KEY: str
SECRET_KEY: str = "3db8542397edddbd6162ad823157e36f8d47232aa646725d4799266229ba7aa4"
ALGORITHM: str = "HS256"
ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
REFRESH_TOKEN_EXPIRE_DAYS: int = 7
# Database
POSTGRES_USER: str
POSTGRES_PASSWORD: str
POSTGRES_DB: str
POSTGRES_USER: str = "postgres"
POSTGRES_PASSWORD: str = "postgres"
POSTGRES_DB: str = "newplanet"
POSTGRES_HOST: str = "localhost"
POSTGRES_PORT: int = 5432
DATABASE_URL: Optional[str] = None
@@ -43,15 +43,15 @@ class Settings(BaseSettings):
# Storage (MinIO/S3)
STORAGE_ENDPOINT: str = "localhost:9000"
STORAGE_ACCESS_KEY: str
STORAGE_SECRET_KEY: str
STORAGE_ACCESS_KEY: str = "minioadmin"
STORAGE_SECRET_KEY: str = "minioadmin"
STORAGE_BUCKET: str = "new-planet-images"
STORAGE_USE_SSL: bool = False
STORAGE_REGION: str = "us-east-1"
# GigaChat
GIGACHAT_CLIENT_ID: str
GIGACHAT_CLIENT_SECRET: str
GIGACHAT_CLIENT_ID: str = "019966f4-1c5c-7382-9006-b84419fbe5d1"
GIGACHAT_CLIENT_SECRET: str = "MDE5OTY2ZjQtMWM1Yy03MzgyLTkwMDYtYjg0NDE5ZmJlNWQxOjJjODBmOWE2LWU4YWMtNDE4YS1iOGVkLWE4NTE0YzVkNDAwNw=="
GIGACHAT_AUTH_URL: str = "https://ngw.devices.sberbank.ru:9443/api/v2/oauth"
GIGACHAT_BASE_URL: str = "https://gigachat.devices.sberbank.ru/api/v1"
GIGACHAT_MODEL_CHAT: str = "GigaChat-2-Lite"

2
new-planet-backend/app/core/security.py
View File

@@ -2,7 +2,7 @@ from datetime import datetime, timedelta
from typing import Optional, Dict, Any
from jose import JWTError, jwt
import bcrypt
from app.core.config import settings
from config import settings
def verify_password(plain_password: str, hashed_password: str) -> bool:

2
new-planet-backend/app/db/base.py
View File

@@ -1,4 +1,4 @@
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import declarative_base
from sqlalchemy import Column, DateTime, func, String
import uuid

4
new-planet-backend/app/db/init_db.py
View File

@@ -1,5 +1,5 @@
from app.db.base import Base
from app.db.session import engine
from base import Base
from session import engine
from app.models import user, schedule, task, reward, ai_conversation

3
new-planet-backend/app/db/session.py
View File

@@ -3,8 +3,7 @@ from app.core.config import settings
engine = create_async_engine(
settings.database_url,
echo=settings.DEBUG,
future=True
echo=settings.DEBUG
)
AsyncSessionLocal = async_sessionmaker(

17
new-planet-backend/app/main.py
View File

@@ -1,3 +1,18 @@
import sys
from pathlib import Path
# Добавляем корневую директорию проекта в PYTHONPATH при прямом запуске
# Это нужно, чтобы Python мог найти модуль 'app'
# Проверяем, запускается ли файл напрямую, проверяя имя скрипта
if sys.argv and len(sys.argv) > 0:
script_path = Path(sys.argv[0]).resolve()
current_file = Path(__file__).resolve()
# Если скрипт запускается напрямую (не через модуль)
if script_path == current_file or script_path.name == current_file.name:
project_root = current_file.parent.parent # new-planet-backend
if str(project_root) not in sys.path:
sys.path.insert(0, str(project_root))
from fastapi import FastAPI, Request
from fastapi.exceptions import RequestValidationError
from starlette.exceptions import HTTPException as StarletteHTTPException
@@ -73,7 +88,7 @@ if __name__ == "__main__":
import uvicorn
uvicorn.run(
"app.main:app",
host="0.0.0.0",
host="127.0.0.1",
port=8000,
reload=settings.DEBUG
)

38
new-planet-backend/app/schemas/ai.py
View File

@@ -38,3 +38,41 @@ class ConversationHistory(BaseModel):
class Config:
from_attributes = True
class ConversationListItem(BaseModel):
"""Элемент списка разговоров"""
conversation_id: str
last_message: Optional[str] = None
created_at: datetime
updated_at: datetime
message_count: int = 0
class Config:
from_attributes = True
class ScheduleUpdateRequest(BaseModel):
"""Запрос на обновление расписания через ИИ"""
user_request: str = Field(..., min_length=1, max_length=1000, description="Описание желаемых изменений")
class ScheduleUpdateResponse(BaseModel):
"""Ответ после обновления расписания"""
schedule_id: str
title: str
tasks: List[Dict[str, Any]]
tokens_used: Optional[int] = None
class RecommendationRequest(BaseModel):
"""Запрос на получение рекомендаций"""
preferences: List[str] = Field(default_factory=list, description="Предпочтения пользователя")
category: Optional[str] = Field(None, description="Категория заданий")
completed_tasks: Optional[List[str]] = Field(default_factory=list, description="Уже выполненные задания")
top_k: int = Field(5, ge=1, le=20, description="Количество рекомендаций")
class RecommendationResponse(BaseModel):
"""Ответ с рекомендациями"""
recommendations: List[Dict[str, Any]]
total: int

94
new-planet-backend/app/services/gigachat_service.py
View File

@@ -1,11 +1,17 @@
import os
import aiohttp
import ssl
import base64
import uuid
import time
from urllib.parse import urlencode
from typing import Optional, List, Dict, Any
from dotenv import load_dotenv
from app.core.config import settings
load_dotenv()
class GigaChatService:
def __init__(self):
@@ -19,8 +25,19 @@ class GigaChatService:
if time.time() < (self.token_expires_at - 60):
return self.access_token
credentials = f"{settings.GIGACHAT_CLIENT_ID}:{settings.GIGACHAT_CLIENT_SECRET}"
encoded_credentials = base64.b64encode(credentials.encode()).decode()
# Проверяем наличие credentials
client_id = os.getenv("GIGACHAT_CLIENT_ID")
client_secret = os.getenv("GIGACHAT_CLIENT_SECRET")
if not client_id or not client_secret:
raise Exception(
"GigaChat credentials not configured. "
"Please set GIGACHAT_CLIENT_ID and GIGACHAT_CLIENT_SECRET in .env file"
)
# Формируем credentials и кодируем в Base64 с явным указанием UTF-8
credentials = f"{client_id}:{client_secret}".strip().encode('utf-8') # Обрезаем лишние символы
encoded_credentials = base64.b64encode(credentials).decode('utf-8')
headers = {
"Authorization": f"Basic {encoded_credentials}",
@@ -29,31 +46,61 @@ class GigaChatService:
"RqUID": str(uuid.uuid4())
}
data = {"scope": "GIGACHAT_API_PERS"}
# Правильно кодируем данные формы (как в рабочем примере)
form_data = {
"grant_type": "client_credentials",
"scope": "GIGACHAT_API_PERS"
}
# Создаем SSL контекст без проверки сертификата (только для разработки!)
ssl_context = ssl.create_default_context()
ssl_context.check_hostname = False
ssl_context.verify_mode = ssl.CERT_NONE
connector = aiohttp.TCPConnector(ssl=ssl_context)
# Отключаем проверку SSL (только для разработки!)
# Используем ssl=False для полного отключения проверки сертификата
connector = aiohttp.TCPConnector(ssl=False)
async with aiohttp.ClientSession(connector=connector) as session:
async with session.post(
settings.GIGACHAT_AUTH_URL,
os.getenv("GIGACHAT_BASE_URL"),
headers=headers,
data=data
data=form_data
) as response:
if response.status != 200:
raise Exception(f"Failed to get token: {response.status}")
# Получаем детали ошибки из ответа
try:
error_body = await response.text()
# Пытаемся распарсить как JSON, если не получается - возвращаем текст
try:
error_json = await response.json()
error_detail = error_json.get("error_description") or error_json.get("error") or str(error_json)
except:
error_detail = error_body
except:
error_detail = "No error details available"
raise Exception(
f"Failed to get token: HTTP {response.status}. "
f"Error details: {error_detail}. "
f"Check your GIGACHAT_CLIENT_ID and GIGACHAT_CLIENT_SECRET_2 in .env file"
)
result = await response.json()
self.access_token = result.get("access_token")
expires_in = result.get("expires_at", 1800)
# expires_at может быть timestamp или количество секунд
if expires_in > 1000000000: # Это timestamp
self.token_expires_at = expires_in
else: # Это количество секунд
if not self.access_token:
raise Exception(f"Token not found in response: {result}")
# Обрабатываем время истечения токена (может быть expires_at или expires_in)
expires_at = result.get("expires_at")
expires_in = result.get("expires_in")
if expires_at:
# expires_at может быть timestamp или количество секунд
if expires_at > 1000000000: # Это timestamp
self.token_expires_at = expires_at
else: # Это количество секунд
self.token_expires_at = time.time() + expires_at
elif expires_in:
# expires_in - это всегда количество секунд до истечения
self.token_expires_at = time.time() + expires_in
else:
# По умолчанию 30 минут (1800 секунд)
self.token_expires_at = time.time() + 1800
return self.access_token
@@ -65,7 +112,7 @@ class GigaChatService:
) -> Dict[str, Any]:
"""Отправить сообщение в GigaChat"""
token = await self._get_token()
model = model or settings.GIGACHAT_MODEL_CHAT
model = model or settings.GIGACHAT_MODEL_CHAT or "GigaChat"
messages = context or []
messages.append({"role": "user", "content": message})
@@ -82,12 +129,9 @@ class GigaChatService:
"max_tokens": 2000
}
# Создаем SSL контекст без проверки сертификата (только для разработки!)
ssl_context = ssl.create_default_context()
ssl_context.check_hostname = False
ssl_context.verify_mode = ssl.CERT_NONE
connector = aiohttp.TCPConnector(ssl=ssl_context)
# Отключаем проверку SSL (только для разработки!)
# Используем ssl=False для полного отключения проверки сертификата
connector = aiohttp.TCPConnector(ssl=False)
async with aiohttp.ClientSession(connector=connector) as session:
async with session.post(
f"{settings.GIGACHAT_BASE_URL}/chat/completions",