FIx typo
This commit is contained in:
parent
f6122d7091
commit
cd67e88e6a
|
@ -1,4 +1,3 @@
|
||||||
import time
|
|
||||||
from fastapi import APIRouter, HTTPException
|
from fastapi import APIRouter, HTTPException
|
||||||
|
|
||||||
from src.models.lead import LeadInput
|
from src.models.lead import LeadInput
|
||||||
|
@ -11,8 +10,6 @@ router = APIRouter(prefix="/api/v1", tags=["generation"])
|
||||||
|
|
||||||
@router.post("/generate_email", response_model=EmailResponse)
|
@router.post("/generate_email", response_model=EmailResponse)
|
||||||
async def generate_email(lead_data: LeadInput):
|
async def generate_email(lead_data: LeadInput):
|
||||||
start_time = time.time()
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
request = lead_data.model_dump()
|
request = lead_data.model_dump()
|
||||||
|
|
||||||
|
|
|
@ -13,7 +13,7 @@ async def health_check():
|
||||||
chroma_store = ChromaStore()
|
chroma_store = ChromaStore()
|
||||||
doc_count = chroma_store.get_count()
|
doc_count = chroma_store.get_count()
|
||||||
|
|
||||||
embedding_service = EmbeddingService()
|
EmbeddingService()
|
||||||
|
|
||||||
health_status = {
|
health_status = {
|
||||||
"status": "healthy",
|
"status": "healthy",
|
||||||
|
|
|
@ -98,13 +98,6 @@ class DocumentChunker:
|
||||||
current_section = ""
|
current_section = ""
|
||||||
current_title = "Введение"
|
current_title = "Введение"
|
||||||
|
|
||||||
semantic_markers = [
|
|
||||||
("Проблема", ["проблема", "вызов", "трудност", "сложност"]),
|
|
||||||
("Решение", ["решение", "как", "что сделали", "подход"]),
|
|
||||||
("Результат", ["результат", "итог", "достижени", "эффект"]),
|
|
||||||
("О клиенте", ["о клиенте", "о компании", "клиент"]),
|
|
||||||
]
|
|
||||||
|
|
||||||
for line in lines:
|
for line in lines:
|
||||||
line_stripped = line.strip()
|
line_stripped = line.strip()
|
||||||
|
|
||||||
|
|
|
@ -21,8 +21,6 @@ class MarkdownLoader:
|
||||||
with open(file_path, "r", encoding="utf-8") as f:
|
with open(file_path, "r", encoding="utf-8") as f:
|
||||||
content = f.read()
|
content = f.read()
|
||||||
|
|
||||||
html_content = self.md.convert(content)
|
|
||||||
|
|
||||||
title = self._extract_title(content)
|
title = self._extract_title(content)
|
||||||
doc_id = self._generate_doc_id(file_path)
|
doc_id = self._generate_doc_id(file_path)
|
||||||
|
|
||||||
|
|
|
@ -217,5 +217,5 @@ class TestEmbeddingService:
|
||||||
with patch("src.app.config.settings") as mock_settings:
|
with patch("src.app.config.settings") as mock_settings:
|
||||||
mock_settings.llm_provider = "openai"
|
mock_settings.llm_provider = "openai"
|
||||||
|
|
||||||
service = EmbeddingService()
|
EmbeddingService()
|
||||||
mock_provider.assert_called_once()
|
mock_provider.assert_called_once()
|
||||||
|
|
Loading…
Reference in New Issue