62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
from openai import OpenAI
|
||
from ..config import OPENAI_API_KEY, OPENAI_BASE_URL, DEFAULT_MODEL
|
||
from ..utils.logger import log
|
||
|
||
class AIInterface:
|
||
def __init__(self):
|
||
self.client = OpenAI(
|
||
api_key=OPENAI_API_KEY,
|
||
base_url=OPENAI_BASE_URL,
|
||
)
|
||
|
||
def send_request(self, prompt: str, context: dict = None) -> dict:
|
||
"""
|
||
Отправляет запрос на эндпоинт ИИ через OpenAI и возвращает ответ.
|
||
"""
|
||
try:
|
||
log(f"Sending request to AI endpoint with prompt: {prompt}")
|
||
|
||
# Подготовка сообщений для чат-комплитшена
|
||
messages = [{"role": "user", "content": prompt}]
|
||
# if context:
|
||
# # Преобразуем контекст в строку
|
||
# context_str = self._format_context(context)
|
||
# messages.append({"role": "system", "content": context_str})
|
||
|
||
# Создание запроса к OpenAI
|
||
response = self.client.chat.completions.create(
|
||
model=DEFAULT_MODEL,
|
||
messages=messages,
|
||
stream=False, # Пока отключаем стриминг для простоты
|
||
)
|
||
|
||
# Обработка ответа
|
||
return {"response": response.choices[0].message.content}
|
||
|
||
except Exception as e:
|
||
log(f"Error communicating with AI endpoint: {e}", level="error")
|
||
raise
|
||
|
||
def _format_context(self, context: dict) -> str:
|
||
"""
|
||
Преобразует контекст в строку для передачи в OpenAI API.
|
||
"""
|
||
formatted_context = []
|
||
for key, value in context.items():
|
||
formatted_context.append(f"{key}: {value}")
|
||
return "\n".join(formatted_context)
|
||
|
||
|
||
def analyze_project(self, project_structure: str, prompt: str) -> dict:
|
||
"""
|
||
Анализирует структуру проекта и запрос пользователя через ИИ.
|
||
"""
|
||
context = {"project_structure": project_structure}
|
||
return self.send_request(prompt, context)
|
||
|
||
def request_additional_info(self, message: str) -> str:
|
||
"""
|
||
Запрашивает дополнительную информацию у пользователя.
|
||
"""
|
||
log("Requesting additional information from user.")
|
||
return input(f"{message}\n> ") |