fix: improve maintenance and synchronize live conversations

This commit is contained in:
WangShen007 committed 2026-10-29 09:46:21 +08:00
1 parent bb541bd212
commit ef86a5d3af
297 files changed
+2983 -1274

No files matched your search

+3 -9
View File
@@ -14,8 +14,8 @@ See the License for the specific language governing permissions and
limitations under the License.
"""
from agno.knowledge.pdf import PDFKnowledgeBase, PDFImageReader, PDFReader
from agno.vectordb.pgvector import PgVector, SearchType
from agno.knowledge.pdf import PDFImageReader, PDFReader
from agno.vectordb.pgvector import SearchType
from app.knowledge.optimized_pgvector import OptimizedPgVector
from app.core.config import settings
from app.core.logger import get_logger
@@ -31,7 +31,6 @@ from app.knowledge.pdf_limits import PdfRefused, check_page_count
from app.models.knowledge import Knowledge, SourceType
from app.models.knowledge_to_agent import KnowledgeToAgent
from app.models.knowledge_queue import ProcessingStage, QueueStatus
from app.repositories.ai_config import AIConfigRepository
from app.database import SessionLocal
from app.repositories.knowledge_to_agent import KnowledgeToAgentRepository
from app.repositories.knowledge import KnowledgeRepository
@@ -47,7 +46,7 @@ from app.knowledge.embedder import get_embedder
# Try to import enterprise modules
try:
from app.enterprise.repositories.subscription import SubscriptionRepository
from app.enterprise.repositories.subscription import SubscriptionRepository # noqa: F401 - class presence selects enterprise knowledge behavior
HAS_ENTERPRISE = True
except ImportError:
HAS_ENTERPRISE = False
@@ -81,11 +80,6 @@ class KnowledgeManager:
self.org_id = org_id
self.agent_id = agent_id
# Get API key from AI config using context manager
with SessionLocal() as db:
ai_config_repo = AIConfigRepository(db)
ai_config = ai_config_repo.get_active_config(org_id)
# Default to SentenceTransformer embedder if no AI config is found
embedder = None
table_name = f"d_{org_id}"