Add pgvector and text service containerization

MaskerPRC · Sep 24, 2024 · 0ca32ad · 0ca32ad
1 parent d3e06f4
commit 0ca32ad
Show file tree

Hide file tree

Showing 7 changed files with 2 additions and 148 deletions.
diff --git a/.env-example b/.env-example
@@ -24,19 +24,3 @@ SPECBOT_AI_MODEL=gpt-4-0613
 REPAIR_PRO_AI_MODEL=claude-3-5-sonnet-20240620
 OPENAI_API_KEY=*** Provide your API Key ***
 OPENAI_BASE_URL=*** Provide your Base url ***
-
-# 数据库配置
-DB_NAME=
-DB_USER=
-DB_PASSWORD=
-DB_HOST=
-DB_PORT=
-
-# 模型名称配置
-MODEL_NAME=
-
-# 配置项
-VECTOR_EXTENSION=
-TABLE_NAME=
-VECTOR_DIMENSION=
-LANGUAGE=
diff --git a/infra_ai_service/api/ai_enhance/embedding.py b/infra_ai_service/api/ai_enhance/embedding.py
@@ -1,11 +1,7 @@
 from fastapi import APIRouter
 
-from infra_ai_service.api.ai_enhance.text_process import TextInput
-from infra_ai_service.model.model import EmbeddingOutput
-from infra_ai_service.service.embedding_service import (
-    create_embedding,
-    get_collection_status,
-)
+from infra_ai_service.model.model import EmbeddingOutput, TextInput
+from infra_ai_service.service.embedding_service import create_embedding
 
 router = APIRouter()
 

diff --git a/infra_ai_service/api/ai_enhance/text_process.py b/infra_ai_service/api/ai_enhance/text_process.py
diff --git a/infra_ai_service/api/router.py b/infra_ai_service/api/router.py
@@ -15,7 +15,6 @@
 api_router.include_router(
     spec_repair_process, prefix="/spec-repair", tags=["repair"]
 )
-api_router.include_router(text_process_router, prefix="/text", tags=["text"])
 api_router.include_router(
     embedding_router, prefix="/embedding", tags=["embedding"]
 )

diff --git a/infra_ai_service/service/text_service.py b/infra_ai_service/service/text_service.py
diff --git a/tests/test_all.py b/tests/test_all.py
diff --git a/tests/test_api.py b/tests/test_api.py