From f6122d7091fa3edab90fcf7dd3dd6c0bad814f8d Mon Sep 17 00:00:00 2001 From: itqop Date: Fri, 18 Jul 2025 20:26:17 +0300 Subject: [PATCH] Fix typo --- src/app/routers/generate.py | 5 ++--- src/graph/build_graph.py | 1 - src/graph/nodes/input_validation.py | 1 - src/graph/nodes/parse_output.py | 2 +- src/ingest/chunker.py | 2 +- src/ingest/ingest_cli.py | 4 ++-- src/llm_clients/base.py | 1 - src/llm_clients/gemini_client.py | 1 - src/llm_clients/openai_client.py | 1 - src/services/embeddings.py | 2 +- src/services/prompt_templates.py | 1 - src/services/retrieval.py | 4 ++-- src/tests/test_api.py | 2 -- src/tests/test_ingest.py | 4 ---- 14 files changed, 9 insertions(+), 22 deletions(-) diff --git a/src/app/routers/generate.py b/src/app/routers/generate.py index e7cdc81..4520b92 100644 --- a/src/app/routers/generate.py +++ b/src/app/routers/generate.py @@ -1,9 +1,8 @@ import time -from fastapi import APIRouter, HTTPException, Depends -from typing import Dict, Any +from fastapi import APIRouter, HTTPException from src.models.lead import LeadInput -from src.models.email import EmailResponse, ErrorResponse +from src.models.email import EmailResponse from src.graph.build_graph import email_generation_graph from src.app.config import settings diff --git a/src/graph/build_graph.py b/src/graph/build_graph.py index 9c3a9bb..cece6fe 100644 --- a/src/graph/build_graph.py +++ b/src/graph/build_graph.py @@ -1,5 +1,4 @@ from langgraph.graph import StateGraph, END -from typing import Dict, Any from src.graph.state import EmailGenerationState from src.graph.nodes.input_validation import input_validation_node diff --git a/src/graph/nodes/input_validation.py b/src/graph/nodes/input_validation.py index 15b56dc..3df97c8 100644 --- a/src/graph/nodes/input_validation.py +++ b/src/graph/nodes/input_validation.py @@ -1,4 +1,3 @@ -from typing import Dict, Any from src.graph.state import EmailGenerationState from src.models.lead import LeadInput, LeadModel from src.models.errors import ValidationError diff --git a/src/graph/nodes/parse_output.py b/src/graph/nodes/parse_output.py index 24ea513..d96b9bb 100644 --- a/src/graph/nodes/parse_output.py +++ b/src/graph/nodes/parse_output.py @@ -19,7 +19,7 @@ def parse_output_node(state: EmailGenerationState) -> EmailGenerationState: try: parsed_data = json.loads(content) - except json.JSONDecodeError as e: + except json.JSONDecodeError: parsed_data = _fallback_parse(content) if not isinstance(parsed_data, dict): diff --git a/src/ingest/chunker.py b/src/ingest/chunker.py index 8e78b6c..d45af89 100644 --- a/src/ingest/chunker.py +++ b/src/ingest/chunker.py @@ -1,6 +1,6 @@ import re import tiktoken -from typing import List, Dict, Any +from typing import List from dataclasses import dataclass from src.ingest.loader import Document diff --git a/src/ingest/ingest_cli.py b/src/ingest/ingest_cli.py index f43c4eb..ebbee9d 100644 --- a/src/ingest/ingest_cli.py +++ b/src/ingest/ingest_cli.py @@ -2,7 +2,7 @@ import os import sys import argparse import hashlib -from typing import List, Dict, Any +from typing import Dict, Any sys.path.append( os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) @@ -31,7 +31,7 @@ def run_ingest( embedding_model: str = None, ) -> Dict[str, Any]: - print(f"Starting ingestion process...") + print("Starting ingestion process...") print(f"Data directory: {data_dir}") print(f"Persist directory: {persist_dir or settings.chroma_persist_dir}") print(f"Recreate: {recreate}") diff --git a/src/llm_clients/base.py b/src/llm_clients/base.py index 5d05ae9..4a3b51e 100644 --- a/src/llm_clients/base.py +++ b/src/llm_clients/base.py @@ -1,5 +1,4 @@ from abc import ABC, abstractmethod -from typing import List, Dict, Any, Optional, Tuple from src.models.email import LLMRawOutput diff --git a/src/llm_clients/gemini_client.py b/src/llm_clients/gemini_client.py index a39a6b9..175ebb9 100644 --- a/src/llm_clients/gemini_client.py +++ b/src/llm_clients/gemini_client.py @@ -1,6 +1,5 @@ import google.generativeai as genai import tiktoken -from typing import Dict, Any from src.llm_clients.base import LLMClient from src.models.email import LLMRawOutput diff --git a/src/llm_clients/openai_client.py b/src/llm_clients/openai_client.py index 949d60a..f139c13 100644 --- a/src/llm_clients/openai_client.py +++ b/src/llm_clients/openai_client.py @@ -1,6 +1,5 @@ import openai import tiktoken -from typing import Dict, Any from src.llm_clients.base import LLMClient from src.models.email import LLMRawOutput diff --git a/src/services/embeddings.py b/src/services/embeddings.py index 0d5d28f..4c5055c 100644 --- a/src/services/embeddings.py +++ b/src/services/embeddings.py @@ -1,4 +1,4 @@ -from typing import List, Union +from typing import List from abc import ABC, abstractmethod import openai import google.generativeai as genai diff --git a/src/services/prompt_templates.py b/src/services/prompt_templates.py index fa265eb..3d2fc6a 100644 --- a/src/services/prompt_templates.py +++ b/src/services/prompt_templates.py @@ -1,4 +1,3 @@ -from typing import Dict, Any from src.models.lead import LeadModel from src.models.email import RankedContext from src.app.config import settings diff --git a/src/services/retrieval.py b/src/services/retrieval.py index 3b3a145..8d8822f 100644 --- a/src/services/retrieval.py +++ b/src/services/retrieval.py @@ -1,10 +1,10 @@ -from typing import List, Dict, Any, Optional +from typing import List, Optional import tiktoken import hashlib import time from src.models.email import DocChunk, RetrievalQuery, RankedContext -from src.models.lead import LeadModel, LeadFeatures +from src.models.lead import LeadFeatures from src.services.chroma_store import ChromaStore from src.services.embeddings import EmbeddingService from src.app.config import settings diff --git a/src/tests/test_api.py b/src/tests/test_api.py index 94b120a..7797084 100644 --- a/src/tests/test_api.py +++ b/src/tests/test_api.py @@ -1,5 +1,3 @@ -import pytest -import json from fastapi.testclient import TestClient import sys import os diff --git a/src/tests/test_ingest.py b/src/tests/test_ingest.py index 46b7f5f..05db840 100644 --- a/src/tests/test_ingest.py +++ b/src/tests/test_ingest.py @@ -1,7 +1,5 @@ -import pytest import tempfile import os -import shutil import sys from unittest.mock import Mock, patch @@ -140,7 +138,6 @@ class TestDocumentChunker: class TestLeadModelExtraction: def test_role_category_extraction(self): - from src.models.lead import LeadModel assert LeadModel._extract_role_category("Технический директор") == "tech" assert LeadModel._extract_role_category("Финансовый директор") == "finance" @@ -148,7 +145,6 @@ class TestLeadModelExtraction: assert LeadModel._extract_role_category("HR менеджер") == "hr" def test_industry_tag_extraction(self): - from src.models.lead import LeadModel assert ( LeadModel._extract_industry_tag("маркетинговое агентство")