From d8d40997b8ca018ea2e73c4c3c6d691ab332dfd2 Mon Sep 17 00:00:00 2001 From: renzhiyuan <465386466@qq.com> Date: Thu, 28 Aug 2025 18:04:11 +0800 Subject: [PATCH] =?UTF-8?q?[+]=E5=88=A0=E9=99=A4=E9=93=BE=E8=B7=AF?= =?UTF-8?q?=E8=BF=BD=E8=B8=AA=EF=BC=8C=E4=BF=AE=E6=94=B9=E9=A1=B9=E7=9B=AE?= =?UTF-8?q?=E5=90=8D=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 6 +- README_EN.md | 6 +- client/README.md | 2 +- client/README_EN.md | 2 +- client/go.mod | 2 +- cmd/server/main.go | 10 +- docker-compose.bak | 198 ++++++++++++++++++ docker-compose.yml | 23 -- docs/API.md | 8 +- go.mod | 4 +- go.sum | 6 - internal/application/repository/chunk.go | 6 +- internal/application/repository/knowledge.go | 4 +- .../application/repository/knowledgebase.go | 5 +- internal/application/repository/message.go | 4 +- internal/application/repository/model.go | 4 +- .../retriever/elasticsearch/structs.go | 2 +- .../retriever/elasticsearch/v7/repository.go | 10 +- .../retriever/elasticsearch/v8/repository.go | 10 +- .../retriever/postgres/repository.go | 8 +- .../repository/retriever/postgres/structs.go | 4 +- internal/application/repository/session.go | 4 +- internal/application/repository/tenant.go | 6 +- .../service/chat_pipline/chat_completion.go | 6 +- .../chat_pipline/chat_completion_stream.go | 6 +- .../service/chat_pipline/chat_pipline.go | 2 +- .../service/chat_pipline/chat_pipline_test.go | 2 +- .../service/chat_pipline/common.go | 8 +- .../service/chat_pipline/filter_top_k.go | 4 +- .../service/chat_pipline/into_chat_message.go | 4 +- .../application/service/chat_pipline/merge.go | 4 +- .../service/chat_pipline/preprocess.go | 8 +- .../service/chat_pipline/rerank.go | 8 +- .../service/chat_pipline/rewrite.go | 10 +- .../service/chat_pipline/search.go | 8 +- .../service/chat_pipline/stream_filter.go | 4 +- .../service/chat_pipline/tracing.go | 4 +- internal/application/service/chunk.go | 6 +- internal/application/service/common_test.go | 27 +++ internal/application/service/dataset.go | 6 +- internal/application/service/evaluation.go | 8 +- internal/application/service/file/cos.go | 2 +- internal/application/service/file/dummy.go | 2 +- internal/application/service/file/local.go | 4 +- internal/application/service/file/minio.go | 2 +- internal/application/service/graph.go | 12 +- internal/application/service/knowledge.go | 22 +- internal/application/service/knowledgebase.go | 12 +- .../application/service/knowledgebase_test.go | 37 ++++ internal/application/service/message.go | 6 +- internal/application/service/metric/bleu.go | 2 +- internal/application/service/metric/common.go | 2 +- internal/application/service/metric/map.go | 2 +- .../application/service/metric/map_test.go | 2 +- internal/application/service/metric/mrr.go | 2 +- .../application/service/metric/mrr_test.go | 2 +- internal/application/service/metric/ndcg.go | 2 +- .../application/service/metric/precision.go | 2 +- .../service/metric/precision_test.go | 2 +- internal/application/service/metric/recall.go | 2 +- .../application/service/metric/recall_test.go | 2 +- internal/application/service/metric/rouge.go | 2 +- internal/application/service/metric_hook.go | 8 +- internal/application/service/model.go | 14 +- .../service/retriever/composite.go | 14 +- .../keywords_vector_hybrid_indexer.go | 10 +- .../application/service/retriever/registry.go | 4 +- internal/application/service/session.go | 14 +- internal/application/service/tenant.go | 18 +- internal/application/service/test_data.go | 16 +- internal/common/asyncq.go | 2 +- internal/config/config.go | 8 +- internal/container/cleanup.go | 4 +- internal/container/container.go | 38 ++-- internal/handler/chunk.go | 10 +- internal/handler/evaluation.go | 8 +- internal/handler/initialization.go | 18 +- internal/handler/knowledge.go | 8 +- internal/handler/knowledgebase.go | 8 +- internal/handler/message.go | 6 +- internal/handler/model.go | 10 +- internal/handler/session.go | 12 +- internal/handler/tenant.go | 8 +- internal/handler/test_data.go | 8 +- internal/logger/logger.go | 2 +- internal/middleware/auth.go | 6 +- internal/middleware/error_handler.go | 2 +- internal/middleware/logger.go | 4 +- internal/middleware/trace.go | 4 +- internal/models/chat/chat.go | 6 +- internal/models/chat/ollama.go | 6 +- internal/models/chat/remote_api.go | 2 +- internal/models/embedding/batch.go | 2 +- internal/models/embedding/embedder.go | 6 +- internal/models/embedding/ollama.go | 4 +- internal/models/embedding/openai.go | 2 +- internal/models/rerank/remote_api.go | 2 +- internal/models/rerank/reranker.go | 2 +- internal/models/utils/ollama/ollama.go | 2 +- internal/router/router.go | 10 +- internal/stream/factory.go | 2 +- internal/stream/memory_manager.go | 4 +- internal/stream/redis_manager.go | 4 +- internal/types/interfaces/chunk.go | 2 +- internal/types/interfaces/evaluation.go | 2 +- internal/types/interfaces/knowledge.go | 2 +- internal/types/interfaces/knowledgebase.go | 2 +- internal/types/interfaces/message.go | 2 +- internal/types/interfaces/model.go | 8 +- internal/types/interfaces/resource.go | 2 +- internal/types/interfaces/retriever.go | 4 +- internal/types/interfaces/session.go | 2 +- internal/types/interfaces/stream_manager.go | 2 +- internal/types/interfaces/tenant.go | 2 +- scripts/start_all.sh | 2 - services/docreader/src/client/client.go | 2 +- services/docreader/src/client/client_test.go | 2 +- services/docreader/src/proto/docreader.pb.go | 4 +- services/docreader/src/proto/docreader.proto | 2 +- .../docreader/src/proto/docreader_grpc.pb.go | 2 +- .../docreader/src/proto/docreader_pb2_grpc.py | 2 +- 121 files changed, 592 insertions(+), 354 deletions(-) create mode 100644 docker-compose.bak create mode 100644 internal/application/service/common_test.go create mode 100644 internal/application/service/knowledgebase_test.go diff --git a/README.md b/README.md index c3e01ff..f78bb5c 100644 --- a/README.md +++ b/README.md @@ -11,7 +11,7 @@ 微信对话开放平台 - + License

@@ -93,7 +93,7 @@ WeKnora 采用现代化模块化设计，构建了一条完整的文档理解与 ```bash # 克隆主仓库 -git clone https://github.com/Tencent/WeKnora.git +git clone https://knowlege-lsxd.git cd WeKnora ``` @@ -240,7 +240,7 @@ make clean-db ## 🤝 贡献指南 -我们欢迎社区用户参与贡献！如有建议、Bug 或新功能需求，请通过 [Issue](https://github.com/Tencent/WeKnora/issues) 提出，或直接提交 Pull Request。 +我们欢迎社区用户参与贡献！如有建议、Bug 或新功能需求，请通过 [Issue](https://knowlege-lsxd/issues) 提出，或直接提交 Pull Request。 ### 🎯 贡献方式 diff --git a/README_EN.md b/README_EN.md index 04904b1..a4d1fbc 100644 --- a/README_EN.md +++ b/README_EN.md @@ -11,7 +11,7 @@ 微信对话开放平台 - + License

@@ -93,7 +93,7 @@ Make sure the following tools are installed on your system: ```bash # Clone the main repository -git clone https://github.com/Tencent/WeKnora.git +git clone https://knowlege-lsxd.git cd WeKnora ``` @@ -206,7 +206,7 @@ make clean-db ## 🤝 Contributing -We welcome community contributions! For suggestions, bugs, or feature requests, please submit an [Issue](https://github.com/Tencent/WeKnora/issues) or directly create a Pull Request. +We welcome community contributions! For suggestions, bugs, or feature requests, please submit an [Issue](https://knowlege-lsxd/issues) or directly create a Pull Request. ### 🎯 How to Contribute diff --git a/client/README.md b/client/README.md index b236055..5c06c85 100644 --- a/client/README.md +++ b/client/README.md @@ -22,7 +22,7 @@ ```go import ( "context" - "github.com/Tencent/WeKnora/internal/client" + "knowlege-lsxd/internal/client" "time" ) diff --git a/client/README_EN.md b/client/README_EN.md index 99ff39d..c411ed1 100644 --- a/client/README_EN.md +++ b/client/README_EN.md @@ -23,7 +23,7 @@ The client includes the following main functional modules: ```go import ( "context" - "github.com/Tencent/WeKnora/internal/client" + "knowlege-lsxd/internal/client" "time" ) diff --git a/client/go.mod b/client/go.mod index 1a20c4b..cd97f8e 100644 --- a/client/go.mod +++ b/client/go.mod @@ -1,3 +1,3 @@ -module github.com/Tencent/WeKnora/client +module knowlege-lsxd/client go 1.24.2 diff --git a/cmd/server/main.go b/cmd/server/main.go index 24130db..ff210cd 100644 --- a/cmd/server/main.go +++ b/cmd/server/main.go @@ -14,11 +14,11 @@ import ( "github.com/gin-gonic/gin" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/container" - "github.com/Tencent/WeKnora/internal/runtime" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/container" + "knowlege-lsxd/internal/runtime" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types/interfaces" ) func main() { diff --git a/docker-compose.bak b/docker-compose.bak new file mode 100644 index 0000000..ee673e3 --- /dev/null +++ b/docker-compose.bak @@ -0,0 +1,198 @@ +services: + app: + image: wechatopenai/weknora-app:latest + container_name: WeKnora-app + ports: + - "8080:8080" + volumes: + - data-files:/data/files + - ./config:/app/config + environment: + - GIN_MODE=${GIN_MODE} + - DB_DRIVER=postgres + - DB_HOST=postgres + - DB_PORT=5432 + - DB_USER=${DB_USER} + - DB_PASSWORD=${DB_PASSWORD} + - DB_NAME=${DB_NAME} + - TZ=Asia/Shanghai + - OTEL_EXPORTER_OTLP_ENDPOINT=jaeger:4317 + - OTEL_SERVICE_NAME=WeKnora + - OTEL_TRACES_EXPORTER=otlp + - OTEL_METRICS_EXPORTER=none + - OTEL_LOGS_EXPORTER=none + - OTEL_PROPAGATORS=tracecontext,baggage + - RETRIEVE_DRIVER=${RETRIEVE_DRIVER} + - ELASTICSEARCH_ADDR=${ELASTICSEARCH_ADDR} + - ELASTICSEARCH_USERNAME=${ELASTICSEARCH_USERNAME} + - ELASTICSEARCH_PASSWORD=${ELASTICSEARCH_PASSWORD} + - ELASTICSEARCH_INDEX=${ELASTICSEARCH_INDEX} + - DOCREADER_ADDR=docreader:50051 + - STORAGE_TYPE=${STORAGE_TYPE} + - LOCAL_STORAGE_BASE_DIR=${LOCAL_STORAGE_BASE_DIR} + - MINIO_ENDPOINT=minio:9000 + - MINIO_ACCESS_KEY_ID=${MINIO_ACCESS_KEY_ID:-minioadmin} + - MINIO_SECRET_ACCESS_KEY=${MINIO_SECRET_ACCESS_KEY:-minioadmin} + - MINIO_BUCKET_NAME=${MINIO_BUCKET_NAME} + - OLLAMA_BASE_URL=${OLLAMA_BASE_URL:-http://host.docker.internal:11434} + - STREAM_MANAGER_TYPE=${STREAM_MANAGER_TYPE} + - REDIS_ADDR=redis:6379 + - REDIS_PASSWORD=${REDIS_PASSWORD} + - REDIS_DB=${REDIS_DB} + - REDIS_PREFIX=${REDIS_PREFIX} + - ENABLE_GRAPH_RAG=${ENABLE_GRAPH_RAG} + - TENANT_AES_KEY=${TENANT_AES_KEY} + - CONCURRENCY_POOL_SIZE=${CONCURRENCY_POOL_SIZE:-5} + - INIT_LLM_MODEL_NAME=${INIT_LLM_MODEL_NAME} + - INIT_LLM_MODEL_BASE_URL=${INIT_LLM_MODEL_BASE_URL} + - INIT_LLM_MODEL_API_KEY=${INIT_LLM_MODEL_API_KEY} + - INIT_EMBEDDING_MODEL_NAME=${INIT_EMBEDDING_MODEL_NAME} + - INIT_EMBEDDING_MODEL_BASE_URL=${INIT_EMBEDDING_MODEL_BASE_URL} + - INIT_EMBEDDING_MODEL_API_KEY=${INIT_EMBEDDING_MODEL_API_KEY} + - INIT_EMBEDDING_MODEL_DIMENSION=${INIT_EMBEDDING_MODEL_DIMENSION} + - INIT_EMBEDDING_MODEL_ID=${INIT_EMBEDDING_MODEL_ID} + - INIT_RERANK_MODEL_NAME=${INIT_RERANK_MODEL_NAME} + - INIT_RERANK_MODEL_BASE_URL=${INIT_RERANK_MODEL_BASE_URL} + - INIT_RERANK_MODEL_API_KEY=${INIT_RERANK_MODEL_API_KEY} + depends_on: + redis: + condition: service_started + postgres: + condition: service_healthy + minio: + condition: service_started + networks: + - WeKnora-network + restart: unless-stopped + extra_hosts: + - "host.docker.internal:host-gateway" + + minio: + image: minio/minio:latest + container_name: WeKnora-minio + ports: + - "${MINIO_PORT:-9000}:9000" + - "${MINIO_CONSOLE_PORT:-9001}:9001" + environment: + - MINIO_ROOT_USER=${MINIO_ACCESS_KEY_ID:-minioadmin} + - MINIO_ROOT_PASSWORD=${MINIO_SECRET_ACCESS_KEY:-minioadmin} + command: server --console-address ":9001" /data + volumes: + - minio_data:/data + healthcheck: + test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"] + interval: 30s + timeout: 20s + retries: 3 + networks: + - WeKnora-network + + frontend: + image: wechatopenai/weknora-ui:latest + container_name: WeKnora-frontend + ports: + - "80:80" + depends_on: + - app + networks: + - WeKnora-network + restart: unless-stopped + + docreader: + image: wechatopenai/weknora-docreader:latest + container_name: WeKnora-docreader + ports: + - "50051:50051" + environment: + - COS_SECRET_ID=${COS_SECRET_ID} + - COS_SECRET_KEY=${COS_SECRET_KEY} + - COS_REGION=${COS_REGION} + - COS_BUCKET_NAME=${COS_BUCKET_NAME} + - COS_APP_ID=${COS_APP_ID} + - COS_PATH_PREFIX=${COS_PATH_PREFIX} + - COS_ENABLE_OLD_DOMAIN=${COS_ENABLE_OLD_DOMAIN} + - VLM_MODEL_BASE_URL=${VLM_MODEL_BASE_URL} + - VLM_MODEL_NAME=${VLM_MODEL_NAME} + - VLM_MODEL_API_KEY=${VLM_MODEL_API_KEY} + - STORAGE_TYPE=${STORAGE_TYPE} + - MINIO_PUBLIC_ENDPOINT=http://localhost:${MINIO_PORT:-9000} + - MINIO_ENDPOINT=minio:9000 + - MINIO_ACCESS_KEY_ID=${MINIO_ACCESS_KEY_ID:-minioadmin} + - MINIO_SECRET_ACCESS_KEY=${MINIO_SECRET_ACCESS_KEY:-minioadmin} + - MINIO_BUCKET_NAME=${MINIO_BUCKET_NAME} + - MINIO_USE_SSL=${MINIO_USE_SSL} + - WEB_PROXY=${WEB_PROXY} + networks: + - WeKnora-network + restart: unless-stopped + extra_hosts: + - "host.docker.internal:host-gateway" + + jaeger: + image: jaegertracing/all-in-one:latest + ports: + - "6831:6831/udp" # Jaeger Thrift接收器 + - "6832:6832/udp" # Jaeger Thrift接收器(Compact) + - "5778:5778" # 配置端口 + - "16686:16686" # Web UI + - "4317:4317" # OTLP gRPC接收器 + - "4318:4318" # OTLP HTTP接收器 + - "14250:14250" # 接收模型端口 + - "14268:14268" # Jaeger HTTP接收器 + - "9411:9411" # Zipkin兼容性端口 + environment: + - COLLECTOR_OTLP_ENABLED=true + - COLLECTOR_ZIPKIN_HOST_PORT=:9411 + volumes: + - jaeger_data:/var/lib/jaeger # 持久化 Jaeger 数据 + networks: + - WeKnora-network + restart: unless-stopped + # 修改的PostgreSQL配置 + postgres: + image: paradedb/paradedb:latest + container_name: WeKnora-postgres + ports: + - "${DB_PORT}:5432" + environment: + - POSTGRES_USER=${DB_USER} + # NOCC:hardcode-password(工具误报) + - POSTGRES_PASSWORD=${DB_PASSWORD} + - POSTGRES_DB=${DB_NAME} + volumes: + - postgres-data:/var/lib/postgresql/data + - ./migrations/paradedb:/docker-entrypoint-initdb.d + networks: + - WeKnora-network + healthcheck: + test: ["CMD-SHELL", "pg_isready -U ${DB_USER}"] + interval: 10s # 增加时间间隔 + timeout: 10s # 增加超时时间 + retries: 3 # 减少重试次数，让失败更快反馈 + start_period: 30s # 给予初始启动更多时间 + restart: unless-stopped + # 添加停机时的优雅退出时间 + stop_grace_period: 1m + + redis: + image: redis:7.0-alpine + container_name: WeKnora-redis + ports: + - "${REDIS_PORT}:6379" + volumes: + - redis_data:/data + command: redis-server --appendonly yes --requirepass ${REDIS_PASSWORD} + restart: always + networks: + - WeKnora-network + +networks: + WeKnora-network: + driver: bridge + +volumes: + postgres-data: + data-files: + jaeger_data: + redis_data: + minio_data: diff --git a/docker-compose.yml b/docker-compose.yml index ee673e3..2b8ea7d 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -16,7 +16,6 @@ services: - DB_PASSWORD=${DB_PASSWORD} - DB_NAME=${DB_NAME} - TZ=Asia/Shanghai - - OTEL_EXPORTER_OTLP_ENDPOINT=jaeger:4317 - OTEL_SERVICE_NAME=WeKnora - OTEL_TRACES_EXPORTER=otlp - OTEL_METRICS_EXPORTER=none @@ -127,27 +126,6 @@ services: restart: unless-stopped extra_hosts: - "host.docker.internal:host-gateway" - - jaeger: - image: jaegertracing/all-in-one:latest - ports: - - "6831:6831/udp" # Jaeger Thrift接收器 - - "6832:6832/udp" # Jaeger Thrift接收器(Compact) - - "5778:5778" # 配置端口 - - "16686:16686" # Web UI - - "4317:4317" # OTLP gRPC接收器 - - "4318:4318" # OTLP HTTP接收器 - - "14250:14250" # 接收模型端口 - - "14268:14268" # Jaeger HTTP接收器 - - "9411:9411" # Zipkin兼容性端口 - environment: - - COLLECTOR_OTLP_ENABLED=true - - COLLECTOR_ZIPKIN_HOST_PORT=:9411 - volumes: - - jaeger_data:/var/lib/jaeger # 持久化 Jaeger 数据 - networks: - - WeKnora-network - restart: unless-stopped # 修改的PostgreSQL配置 postgres: image: paradedb/paradedb:latest @@ -193,6 +171,5 @@ networks: volumes: postgres-data: data-files: - jaeger_data: redis_data: minio_data: diff --git a/docs/API.md b/docs/API.md index ce1e014..c340d0d 100644 --- a/docs/API.md +++ b/docs/API.md @@ -772,7 +772,7 @@ curl --location 'http://localhost:8080/api/v1/knowledge-bases/kb-00000001/knowle --header 'X-API-Key: sk-vQHV2NZI_LK5W7wHQvH3yGYExX8YnhaHwZipUYbiZKCYJbBQ' \ --header 'Content-Type: application/json' \ --data '{ - "url":"https://github.com/Tencent/WeKnora", + "url":"https://knowlege-lsxd", "enable_multimodel":true }' ``` @@ -788,7 +788,7 @@ curl --location 'http://localhost:8080/api/v1/knowledge-bases/kb-00000001/knowle "type": "url", "title": "", "description": "", - "source": "https://github.com/Tencent/WeKnora", + "source": "https://knowlege-lsxd", "parse_status": "processing", "enable_status": "disabled", "embedding_model_id": "dff7bc94-7885-4dd1-bfd5-bd96e4df2fc3", @@ -831,7 +831,7 @@ curl --location 'http://localhost:8080/api/v1/knowledge-bases/kb-00000001/knowle "type": "url", "title": "", "description": "", - "source": "https://github.com/Tencent/WeKnora", + "source": "https://knowlege-lsxd", "parse_status": "pending", "enable_status": "disabled", "embedding_model_id": "dff7bc94-7885-4dd1-bfd5-bd96e4df2fc3", @@ -922,7 +922,7 @@ curl --location 'http://localhost:8080/api/v1/knowledge/batch?ids=9c8af585-ae15- "type": "url", "title": "", "description": "", - "source": "https://github.com/Tencent/WeKnora", + "source": "https://knowlege-lsxd", "parse_status": "pending", "enable_status": "disabled", "embedding_model_id": "dff7bc94-7885-4dd1-bfd5-bd96e4df2fc3", diff --git a/go.mod b/go.mod index 57aa2eb..2eb625d 100644 --- a/go.mod +++ b/go.mod @@ -1,4 +1,4 @@ -module github.com/Tencent/WeKnora +module knowlege-lsxd go 1.24.0 @@ -9,7 +9,6 @@ require ( github.com/elastic/go-elasticsearch/v8 v8.18.0 github.com/gin-contrib/cors v1.7.5 github.com/gin-gonic/gin v1.10.0 - github.com/go-ego/gse v0.80.3 github.com/go-viper/mapstructure/v2 v2.2.1 github.com/google/uuid v1.6.0 github.com/hibiken/asynq v0.25.1 @@ -94,7 +93,6 @@ require ( github.com/subosito/gotenv v1.6.0 // indirect github.com/twitchyliquid64/golang-asm v0.15.1 // indirect github.com/ugorji/go/codec v1.2.12 // indirect - github.com/vcaesar/cedar v0.20.2 // indirect go.opentelemetry.io/auto/sdk v1.1.0 // indirect go.opentelemetry.io/otel/metric v1.37.0 // indirect go.opentelemetry.io/proto/otlp v1.7.0 // indirect diff --git a/go.sum b/go.sum index c3ceeac..81828dd 100644 --- a/go.sum +++ b/go.sum @@ -46,8 +46,6 @@ github.com/gin-contrib/sse v1.0.0 h1:y3bT1mUWUxDpW4JLQg/HnTqV4rozuW4tC9eFKTxYI9E github.com/gin-contrib/sse v1.0.0/go.mod h1:zNuFdwarAygJBht0NTKiSi3jRf6RbqeILZ9Sp6Slhe0= github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU= github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y= -github.com/go-ego/gse v0.80.3 h1:YNFkjMhlhQnUeuoFcUEd1ivh6SOB764rT8GDsEbDiEg= -github.com/go-ego/gse v0.80.3/go.mod h1:Gt3A9Ry1Eso2Kza4MRaiZ7f2DTAvActmETY46Lxg0gU= github.com/go-ini/ini v1.67.0 h1:z6ZrTEZqSWOTyH2FlglNbNgARyHG8oLW9gMELqKr06A= github.com/go-ini/ini v1.67.0/go.mod h1:ByCAeIL28uOIIG0E3PJtZPDL8WnHpFKFOtgjp+3Ies8= github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= @@ -216,10 +214,6 @@ github.com/uptrace/bun/dialect/pgdialect v1.1.12 h1:m/CM1UfOkoBTglGO5CUTKnIKKOAp github.com/uptrace/bun/dialect/pgdialect v1.1.12/go.mod h1:Ij6WIxQILxLlL2frUBxUBOZJtLElD2QQNDcu/PWDHTc= github.com/uptrace/bun/driver/pgdriver v1.1.12 h1:3rRWB1GK0psTJrHwxzNfEij2MLibggiLdTqjTtfHc1w= github.com/uptrace/bun/driver/pgdriver v1.1.12/go.mod h1:ssYUP+qwSEgeDDS1xm2XBip9el1y9Mi5mTAvLoiADLM= -github.com/vcaesar/cedar v0.20.2 h1:TDx7AdZhilKcfE1WvdToTJf5VrC/FXcUOW+KY1upLZ4= -github.com/vcaesar/cedar v0.20.2/go.mod h1:lyuGvALuZZDPNXwpzv/9LyxW+8Y6faN7zauFezNsnik= -github.com/vcaesar/tt v0.20.1 h1:D/jUeeVCNbq3ad8M7hhtB3J9x5RZ6I1n1eZ0BJp7M+4= -github.com/vcaesar/tt v0.20.1/go.mod h1:cH2+AwGAJm19Wa6xvEa+0r+sXDJBT0QgNQey6mwqLeU= github.com/vmihailenco/bufpool v0.1.11 h1:gOq2WmBrq0i2yW5QJ16ykccQ4wH9UyEsgLm6czKAd94= github.com/vmihailenco/bufpool v0.1.11/go.mod h1:AFf/MOy3l2CFTKbxwt0mp2MwnqjNEs5H/UxrkA5jxTQ= github.com/vmihailenco/msgpack/v5 v5.3.5 h1:5gO0H1iULLWGhs2H5tbAHIZTV8/cYafcFOr9znI5mJU= diff --git a/internal/application/repository/chunk.go b/internal/application/repository/chunk.go index 2b01251..5aa8316 100644 --- a/internal/application/repository/chunk.go +++ b/internal/application/repository/chunk.go @@ -4,10 +4,10 @@ import ( "context" "errors" - "github.com/Tencent/WeKnora/internal/common" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "gorm.io/gorm" + "knowlege-lsxd/internal/common" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // chunkRepository implements the ChunkRepository interface diff --git a/internal/application/repository/knowledge.go b/internal/application/repository/knowledge.go index b9b118a..780e542 100644 --- a/internal/application/repository/knowledge.go +++ b/internal/application/repository/knowledge.go @@ -4,9 +4,9 @@ import ( "context" "errors" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "gorm.io/gorm" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) var ErrKnowledgeNotFound = errors.New("knowledge not found") diff --git a/internal/application/repository/knowledgebase.go b/internal/application/repository/knowledgebase.go index efd7ca5..9fc7439 100644 --- a/internal/application/repository/knowledgebase.go +++ b/internal/application/repository/knowledgebase.go @@ -4,8 +4,9 @@ import ( "context" "errors" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" + "gorm.io/gorm" ) diff --git a/internal/application/repository/message.go b/internal/application/repository/message.go index 0808e78..93585b1 100644 --- a/internal/application/repository/message.go +++ b/internal/application/repository/message.go @@ -7,8 +7,8 @@ import ( "gorm.io/gorm" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // messageRepository implements the message repository interface diff --git a/internal/application/repository/model.go b/internal/application/repository/model.go index 568f76c..3e2c6f7 100644 --- a/internal/application/repository/model.go +++ b/internal/application/repository/model.go @@ -4,9 +4,9 @@ import ( "context" "errors" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "gorm.io/gorm" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // modelRepository implements the model repository interface diff --git a/internal/application/repository/retriever/elasticsearch/structs.go b/internal/application/repository/retriever/elasticsearch/structs.go index b30b8bb..db267ad 100644 --- a/internal/application/repository/retriever/elasticsearch/structs.go +++ b/internal/application/repository/retriever/elasticsearch/structs.go @@ -4,7 +4,7 @@ import ( "maps" "slices" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // VectorEmbedding defines the Elasticsearch document structure for vector embeddings diff --git a/internal/application/repository/retriever/elasticsearch/v7/repository.go b/internal/application/repository/retriever/elasticsearch/v7/repository.go index b92c175..03c15ed 100644 --- a/internal/application/repository/retriever/elasticsearch/v7/repository.go +++ b/internal/application/repository/retriever/elasticsearch/v7/repository.go @@ -15,14 +15,14 @@ import ( "os" "strings" - elasticsearchRetriever "github.com/Tencent/WeKnora/internal/application/repository/retriever/elasticsearch" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - typesLocal "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/elastic/go-elasticsearch/v7" "github.com/elastic/go-elasticsearch/v7/esapi" "github.com/google/uuid" + elasticsearchRetriever "knowlege-lsxd/internal/application/repository/retriever/elasticsearch" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + typesLocal "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) type elasticsearchRepository struct { diff --git a/internal/application/repository/retriever/elasticsearch/v8/repository.go b/internal/application/repository/retriever/elasticsearch/v8/repository.go index 17c9883..762bae1 100644 --- a/internal/application/repository/retriever/elasticsearch/v8/repository.go +++ b/internal/application/repository/retriever/elasticsearch/v8/repository.go @@ -6,14 +6,14 @@ import ( "fmt" "os" - elasticsearchRetriever "github.com/Tencent/WeKnora/internal/application/repository/retriever/elasticsearch" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - typesLocal "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/elastic/go-elasticsearch/v8" "github.com/elastic/go-elasticsearch/v8/typedapi/core/search" "github.com/elastic/go-elasticsearch/v8/typedapi/types" + elasticsearchRetriever "knowlege-lsxd/internal/application/repository/retriever/elasticsearch" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + typesLocal "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // elasticsearchRepository implements the RetrieveEngineRepository interface for Elasticsearch v8 diff --git a/internal/application/repository/retriever/postgres/repository.go b/internal/application/repository/retriever/postgres/repository.go index 9d29317..8f53ec1 100644 --- a/internal/application/repository/retriever/postgres/repository.go +++ b/internal/application/repository/retriever/postgres/repository.go @@ -5,13 +5,13 @@ import ( "errors" "fmt" - "github.com/Tencent/WeKnora/internal/common" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/pgvector/pgvector-go" "gorm.io/gorm" "gorm.io/gorm/clause" + "knowlege-lsxd/internal/common" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // pgRepository implements PostgreSQL-based retrieval operations diff --git a/internal/application/repository/retriever/postgres/structs.go b/internal/application/repository/retriever/postgres/structs.go index 45fbc0e..5a7da28 100644 --- a/internal/application/repository/retriever/postgres/structs.go +++ b/internal/application/repository/retriever/postgres/structs.go @@ -6,9 +6,9 @@ import ( "strconv" "time" - "github.com/Tencent/WeKnora/internal/common" - "github.com/Tencent/WeKnora/internal/types" "github.com/pgvector/pgvector-go" + "knowlege-lsxd/internal/common" + "knowlege-lsxd/internal/types" ) // pgVector defines the database model for vector embeddings storage diff --git a/internal/application/repository/session.go b/internal/application/repository/session.go index b0f12fe..373befa 100644 --- a/internal/application/repository/session.go +++ b/internal/application/repository/session.go @@ -4,9 +4,9 @@ import ( "context" "time" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "gorm.io/gorm" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // sessionRepository implements the SessionRepository interface diff --git a/internal/application/repository/tenant.go b/internal/application/repository/tenant.go index 2faf007..c46618d 100644 --- a/internal/application/repository/tenant.go +++ b/internal/application/repository/tenant.go @@ -4,11 +4,11 @@ import ( "context" "errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "gorm.io/gorm" "gorm.io/gorm/clause" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) var ( diff --git a/internal/application/service/chat_pipline/chat_completion.go b/internal/application/service/chat_pipline/chat_completion.go index e589698..cf41835 100644 --- a/internal/application/service/chat_pipline/chat_completion.go +++ b/internal/application/service/chat_pipline/chat_completion.go @@ -3,9 +3,9 @@ package chatpipline import ( "context" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // PluginChatCompletion implements chat completion functionality diff --git a/internal/application/service/chat_pipline/chat_completion_stream.go b/internal/application/service/chat_pipline/chat_completion_stream.go index 7f5644b..639bec7 100644 --- a/internal/application/service/chat_pipline/chat_completion_stream.go +++ b/internal/application/service/chat_pipline/chat_completion_stream.go @@ -3,9 +3,9 @@ package chatpipline import ( "context" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // PluginChatCompletionStream implements streaming chat completion functionality diff --git a/internal/application/service/chat_pipline/chat_pipline.go b/internal/application/service/chat_pipline/chat_pipline.go index c82a7bb..84930dd 100644 --- a/internal/application/service/chat_pipline/chat_pipline.go +++ b/internal/application/service/chat_pipline/chat_pipline.go @@ -3,7 +3,7 @@ package chatpipline import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // Plugin defines the interface for chat pipeline plugins diff --git a/internal/application/service/chat_pipline/chat_pipline_test.go b/internal/application/service/chat_pipline/chat_pipline_test.go index 66b69ab..4838427 100644 --- a/internal/application/service/chat_pipline/chat_pipline_test.go +++ b/internal/application/service/chat_pipline/chat_pipline_test.go @@ -5,7 +5,7 @@ import ( "fmt" "testing" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // Define a test Plugin implementation diff --git a/internal/application/service/chat_pipline/common.go b/internal/application/service/chat_pipline/common.go index 9dff415..2c75e67 100644 --- a/internal/application/service/chat_pipline/common.go +++ b/internal/application/service/chat_pipline/common.go @@ -3,10 +3,10 @@ package chatpipline import ( "context" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // prepareChatModel shared logic to prepare chat model and options diff --git a/internal/application/service/chat_pipline/filter_top_k.go b/internal/application/service/chat_pipline/filter_top_k.go index d57b4b8..f366c95 100644 --- a/internal/application/service/chat_pipline/filter_top_k.go +++ b/internal/application/service/chat_pipline/filter_top_k.go @@ -3,8 +3,8 @@ package chatpipline import ( "context" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" ) // PluginFilterTopK is a plugin that filters search results to keep only the top K items diff --git a/internal/application/service/chat_pipline/into_chat_message.go b/internal/application/service/chat_pipline/into_chat_message.go index 87f5d4e..0a361d9 100644 --- a/internal/application/service/chat_pipline/into_chat_message.go +++ b/internal/application/service/chat_pipline/into_chat_message.go @@ -10,8 +10,8 @@ import ( "strings" "time" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" ) // PluginIntoChatMessage handles the transformation of search results into chat messages diff --git a/internal/application/service/chat_pipline/merge.go b/internal/application/service/chat_pipline/merge.go index dbdde26..a0d36a2 100644 --- a/internal/application/service/chat_pipline/merge.go +++ b/internal/application/service/chat_pipline/merge.go @@ -5,8 +5,8 @@ import ( "encoding/json" "sort" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" ) // PluginMerge handles merging of search result chunks diff --git a/internal/application/service/chat_pipline/preprocess.go b/internal/application/service/chat_pipline/preprocess.go index 20dbf45..e09d6f2 100644 --- a/internal/application/service/chat_pipline/preprocess.go +++ b/internal/application/service/chat_pipline/preprocess.go @@ -7,10 +7,10 @@ import ( "strings" "unicode" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // PluginPreprocess Query preprocessing plugin diff --git a/internal/application/service/chat_pipline/rerank.go b/internal/application/service/chat_pipline/rerank.go index c8ceea2..674204c 100644 --- a/internal/application/service/chat_pipline/rerank.go +++ b/internal/application/service/chat_pipline/rerank.go @@ -7,10 +7,10 @@ import ( "math" "strings" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/rerank" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/rerank" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // PluginRerank implements reranking functionality for chat pipeline diff --git a/internal/application/service/chat_pipline/rewrite.go b/internal/application/service/chat_pipline/rewrite.go index 9c118cb..51732ff 100644 --- a/internal/application/service/chat_pipline/rewrite.go +++ b/internal/application/service/chat_pipline/rewrite.go @@ -11,11 +11,11 @@ import ( "sort" "time" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // PluginRewrite is a plugin for rewriting user queries diff --git a/internal/application/service/chat_pipline/search.go b/internal/application/service/chat_pipline/search.go index 92a019d..ef51a7e 100644 --- a/internal/application/service/chat_pipline/search.go +++ b/internal/application/service/chat_pipline/search.go @@ -4,10 +4,10 @@ import ( "context" "strings" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // PluginSearch implements search functionality for chat pipeline diff --git a/internal/application/service/chat_pipline/stream_filter.go b/internal/application/service/chat_pipline/stream_filter.go index 06d65a5..fc72ee1 100644 --- a/internal/application/service/chat_pipline/stream_filter.go +++ b/internal/application/service/chat_pipline/stream_filter.go @@ -4,8 +4,8 @@ import ( "context" "strings" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" ) // PluginStreamFilter implements stream filtering functionality for chat pipeline diff --git a/internal/application/service/chat_pipline/tracing.go b/internal/application/service/chat_pipline/tracing.go index 8a10e48..8d897a6 100644 --- a/internal/application/service/chat_pipline/tracing.go +++ b/internal/application/service/chat_pipline/tracing.go @@ -6,9 +6,9 @@ import ( "strings" "time" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types" "go.opentelemetry.io/otel/attribute" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types" ) // PluginTracing implements tracing functionality for chat pipeline events diff --git a/internal/application/service/chunk.go b/internal/application/service/chunk.go index b82a4bd..ebbe8d1 100644 --- a/internal/application/service/chunk.go +++ b/internal/application/service/chunk.go @@ -6,9 +6,9 @@ package service import ( "context" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // chunkService implements the ChunkService interface diff --git a/internal/application/service/common_test.go b/internal/application/service/common_test.go new file mode 100644 index 0000000..2ae1f39 --- /dev/null +++ b/internal/application/service/common_test.go @@ -0,0 +1,27 @@ +package service + +import ( + "context" + + "knowlege-lsxd/internal/types" + + "knowlege-lsxd/internal/types/interfaces" + + "github.com/stretchr/testify/mock" +) + +type MockKnowledgeBaseRepo struct { + mock.Mock + interfaces.KnowledgeBaseRepository +} + +func (m *MockKnowledgeBaseRepo) Create(ctx context.Context, kb *types.KnowledgeBase) error { + args := m.Called(ctx, kb) + return args.Error(0) +} + +func must(err error) { + if err != nil { + panic(err) + } +} diff --git a/internal/application/service/dataset.go b/internal/application/service/dataset.go index 259c656..746c47b 100644 --- a/internal/application/service/dataset.go +++ b/internal/application/service/dataset.go @@ -5,10 +5,10 @@ import ( "errors" "fmt" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/parquet-go/parquet-go" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // DatasetService provides operations for working with datasets diff --git a/internal/application/service/evaluation.go b/internal/application/service/evaluation.go index 84c4a47..3bcd6ad 100644 --- a/internal/application/service/evaluation.go +++ b/internal/application/service/evaluation.go @@ -7,12 +7,12 @@ import ( "sync" "time" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/google/uuid" "golang.org/x/sync/errgroup" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) /* diff --git a/internal/application/service/file/cos.go b/internal/application/service/file/cos.go index cae2518..db9611c 100644 --- a/internal/application/service/file/cos.go +++ b/internal/application/service/file/cos.go @@ -10,9 +10,9 @@ import ( "path/filepath" "strings" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/google/uuid" "github.com/tencentyun/cos-go-sdk-v5" + "knowlege-lsxd/internal/types/interfaces" ) // cosFileService implements the FileService interface for Tencent Cloud COS diff --git a/internal/application/service/file/dummy.go b/internal/application/service/file/dummy.go index 3c9e5ce..92d3ec8 100644 --- a/internal/application/service/file/dummy.go +++ b/internal/application/service/file/dummy.go @@ -6,8 +6,8 @@ import ( "io" "mime/multipart" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/google/uuid" + "knowlege-lsxd/internal/types/interfaces" ) // DummyFileService is a no-op implementation of the FileService interface diff --git a/internal/application/service/file/local.go b/internal/application/service/file/local.go index de2c682..e58ce51 100644 --- a/internal/application/service/file/local.go +++ b/internal/application/service/file/local.go @@ -9,8 +9,8 @@ import ( "path/filepath" "time" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types/interfaces" ) // localFileService implements the FileService interface for local file system storage diff --git a/internal/application/service/file/minio.go b/internal/application/service/file/minio.go index af2141f..fcb8d25 100644 --- a/internal/application/service/file/minio.go +++ b/internal/application/service/file/minio.go @@ -7,10 +7,10 @@ import ( "mime/multipart" "path/filepath" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/google/uuid" "github.com/minio/minio-go/v7" "github.com/minio/minio-go/v7/pkg/credentials" + "knowlege-lsxd/internal/types/interfaces" ) // minioFileService MinIO file service implementation diff --git a/internal/application/service/graph.go b/internal/application/service/graph.go index d1271db..c467ac5 100644 --- a/internal/application/service/graph.go +++ b/internal/application/service/graph.go @@ -10,14 +10,14 @@ import ( "sync" "time" - "github.com/Tencent/WeKnora/internal/common" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/models/utils" - "github.com/Tencent/WeKnora/internal/types" "github.com/google/uuid" "golang.org/x/sync/errgroup" + "knowlege-lsxd/internal/common" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/models/utils" + "knowlege-lsxd/internal/types" ) const ( diff --git a/internal/application/service/knowledge.go b/internal/application/service/knowledge.go index bafb669..c7de8f1 100644 --- a/internal/application/service/knowledge.go +++ b/internal/application/service/knowledge.go @@ -17,20 +17,20 @@ import ( "sync" "time" - "github.com/Tencent/WeKnora/internal/application/service/retriever" - "github.com/Tencent/WeKnora/internal/config" - werrors "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/models/utils" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" - "github.com/Tencent/WeKnora/services/docreader/src/client" - "github.com/Tencent/WeKnora/services/docreader/src/proto" "github.com/google/uuid" "go.opentelemetry.io/otel/attribute" "golang.org/x/sync/errgroup" + "knowlege-lsxd/internal/application/service/retriever" + "knowlege-lsxd/internal/config" + werrors "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/models/utils" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" + "knowlege-lsxd/services/docreader/src/client" + "knowlege-lsxd/services/docreader/src/proto" ) // Error definitions for knowledge service operations diff --git a/internal/application/service/knowledgebase.go b/internal/application/service/knowledgebase.go index 19e7a7f..d5ac862 100644 --- a/internal/application/service/knowledgebase.go +++ b/internal/application/service/knowledgebase.go @@ -8,13 +8,13 @@ import ( "encoding/json" - "github.com/Tencent/WeKnora/internal/application/service/retriever" - "github.com/Tencent/WeKnora/internal/common" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/google/uuid" + "knowlege-lsxd/internal/application/service/retriever" + "knowlege-lsxd/internal/common" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // ErrInvalidTenantID represents an error for invalid tenant ID diff --git a/internal/application/service/knowledgebase_test.go b/internal/application/service/knowledgebase_test.go new file mode 100644 index 0000000..ba6658f --- /dev/null +++ b/internal/application/service/knowledgebase_test.go @@ -0,0 +1,37 @@ +package service + +import ( + "testing" + + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/mock" + "knowlege-lsxd/internal/container" + "knowlege-lsxd/internal/runtime" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" +) + +func Test_CreateKnowledge(t *testing.T) { + // 1. 构建容器 + container := container.BuildContainer(runtime.GetContainer()) + + // 2. Mock 依赖 + mockKBRepo := new(MockKnowledgeBaseRepo) + must(container.Provide(func() interfaces.KnowledgeBaseRepository { + return mockKBRepo + })) + + // 3. 解析服务（自动注入 Mock 的 KnowledgeBaseRepository） + var service *knowledgeBaseService + must(container.Invoke(func(k *knowledgeBaseService) { + service = k + })) + + // 4. 设置 Mock 预期 + mockKBRepo.On("Create", context.Background(), mock.Anything).Return(nil) + + // 5. 调用并断言 + base, err := service.CreateKnowledgeBase(context.Background(), &types.KnowledgeBase{Name: "Test"}) + assert.NoError(t, err, base) + mockKBRepo.AssertExpectations(t) +} diff --git a/internal/application/service/message.go b/internal/application/service/message.go index 5304de9..a606ca5 100644 --- a/internal/application/service/message.go +++ b/internal/application/service/message.go @@ -4,9 +4,9 @@ import ( "context" "time" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // messageService implements the MessageService interface for managing messaging operations diff --git a/internal/application/service/metric/bleu.go b/internal/application/service/metric/bleu.go index c1f6dec..f9a4b65 100644 --- a/internal/application/service/metric/bleu.go +++ b/internal/application/service/metric/bleu.go @@ -21,7 +21,7 @@ import ( "math" "strings" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) type BLEUMetric struct { diff --git a/internal/application/service/metric/common.go b/internal/application/service/metric/common.go index 22958c3..e7b5fa9 100644 --- a/internal/application/service/metric/common.go +++ b/internal/application/service/metric/common.go @@ -4,7 +4,7 @@ import ( "regexp" "strings" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) func sum(m map[string]int) int { diff --git a/internal/application/service/metric/map.go b/internal/application/service/metric/map.go index 10f463b..4cfb0f1 100644 --- a/internal/application/service/metric/map.go +++ b/internal/application/service/metric/map.go @@ -1,7 +1,7 @@ package metric import ( - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // MAPMetric calculates Mean Average Precision for retrieval evaluation diff --git a/internal/application/service/metric/map_test.go b/internal/application/service/metric/map_test.go index 7bcaf0e..abf167e 100644 --- a/internal/application/service/metric/map_test.go +++ b/internal/application/service/metric/map_test.go @@ -3,7 +3,7 @@ package metric import ( "testing" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) func TestMAPMetric_Compute(t *testing.T) { diff --git a/internal/application/service/metric/mrr.go b/internal/application/service/metric/mrr.go index 4d89712..ed9fa7a 100644 --- a/internal/application/service/metric/mrr.go +++ b/internal/application/service/metric/mrr.go @@ -1,7 +1,7 @@ package metric import ( - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // MRRMetric calculates Mean Reciprocal Rank for retrieval evaluation diff --git a/internal/application/service/metric/mrr_test.go b/internal/application/service/metric/mrr_test.go index 4d4531d..047e47b 100644 --- a/internal/application/service/metric/mrr_test.go +++ b/internal/application/service/metric/mrr_test.go @@ -3,7 +3,7 @@ package metric import ( "testing" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) func TestMRRMetric_Compute(t *testing.T) { diff --git a/internal/application/service/metric/ndcg.go b/internal/application/service/metric/ndcg.go index 07fd2e1..e67eb86 100644 --- a/internal/application/service/metric/ndcg.go +++ b/internal/application/service/metric/ndcg.go @@ -3,7 +3,7 @@ package metric import ( "math" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // NDCGMetric calculates Normalized Discounted Cumulative Gain diff --git a/internal/application/service/metric/precision.go b/internal/application/service/metric/precision.go index 9121120..f75e52b 100644 --- a/internal/application/service/metric/precision.go +++ b/internal/application/service/metric/precision.go @@ -1,7 +1,7 @@ package metric import ( - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // PrecisionMetric calculates precision for retrieval evaluation diff --git a/internal/application/service/metric/precision_test.go b/internal/application/service/metric/precision_test.go index 4b947f1..099d11b 100644 --- a/internal/application/service/metric/precision_test.go +++ b/internal/application/service/metric/precision_test.go @@ -3,7 +3,7 @@ package metric import ( "testing" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) func TestPrecisionMetric_Compute(t *testing.T) { diff --git a/internal/application/service/metric/recall.go b/internal/application/service/metric/recall.go index 807ad00..7c33b76 100644 --- a/internal/application/service/metric/recall.go +++ b/internal/application/service/metric/recall.go @@ -1,7 +1,7 @@ package metric import ( - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // RecallMetric calculates recall for retrieval evaluation diff --git a/internal/application/service/metric/recall_test.go b/internal/application/service/metric/recall_test.go index e83a54d..354982d 100644 --- a/internal/application/service/metric/recall_test.go +++ b/internal/application/service/metric/recall_test.go @@ -3,7 +3,7 @@ package metric import ( "testing" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) func TestRecallMetric_Compute(t *testing.T) { diff --git a/internal/application/service/metric/rouge.go b/internal/application/service/metric/rouge.go index 22ebe68..3d5b824 100644 --- a/internal/application/service/metric/rouge.go +++ b/internal/application/service/metric/rouge.go @@ -1,6 +1,6 @@ package metric -import "github.com/Tencent/WeKnora/internal/types" +import "knowlege-lsxd/internal/types" // reference: https://github.com/dd-Rebecca/rouge diff --git a/internal/application/service/metric_hook.go b/internal/application/service/metric_hook.go index ec3e80b..20005df 100644 --- a/internal/application/service/metric_hook.go +++ b/internal/application/service/metric_hook.go @@ -4,10 +4,10 @@ import ( "context" "sync" - "github.com/Tencent/WeKnora/internal/application/service/metric" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/application/service/metric" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // MetricList stores and aggregates metric results diff --git a/internal/application/service/model.go b/internal/application/service/model.go index 48cb789..f5b1c0d 100644 --- a/internal/application/service/model.go +++ b/internal/application/service/model.go @@ -4,13 +4,13 @@ import ( "context" "errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/models/rerank" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/models/rerank" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // ErrModelNotFound is returned when a model cannot be found in the repository diff --git a/internal/application/service/retriever/composite.go b/internal/application/service/retriever/composite.go index 701a658..33920be 100644 --- a/internal/application/service/retriever/composite.go +++ b/internal/application/service/retriever/composite.go @@ -8,14 +8,14 @@ import ( "sync" "sync/atomic" - "github.com/Tencent/WeKnora/internal/common" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/runtime" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "go.opentelemetry.io/otel/attribute" + "knowlege-lsxd/internal/common" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/runtime" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // engineInfo holds information about a retrieve engine and its supported retriever types diff --git a/internal/application/service/retriever/keywords_vector_hybrid_indexer.go b/internal/application/service/retriever/keywords_vector_hybrid_indexer.go index 0345994..a181703 100644 --- a/internal/application/service/retriever/keywords_vector_hybrid_indexer.go +++ b/internal/application/service/retriever/keywords_vector_hybrid_indexer.go @@ -5,11 +5,11 @@ import ( "slices" "time" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/models/utils" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/models/utils" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // KeywordsVectorHybridRetrieveEngineService implements a hybrid retrieval engine diff --git a/internal/application/service/retriever/registry.go b/internal/application/service/retriever/registry.go index f489c58..a6475d1 100644 --- a/internal/application/service/retriever/registry.go +++ b/internal/application/service/retriever/registry.go @@ -4,8 +4,8 @@ import ( "fmt" "sync" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // RetrieveEngineRegistry implements the retrieval engine registry diff --git a/internal/application/service/session.go b/internal/application/service/session.go index a229ba3..426e8c2 100644 --- a/internal/application/service/session.go +++ b/internal/application/service/session.go @@ -5,15 +5,15 @@ import ( "errors" "strings" - chatpipline "github.com/Tencent/WeKnora/internal/application/service/chat_pipline" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "go.opentelemetry.io/otel/attribute" "go.opentelemetry.io/otel/codes" + chatpipline "knowlege-lsxd/internal/application/service/chat_pipline" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // sessionService implements the SessionService interface for managing conversation sessions diff --git a/internal/application/service/tenant.go b/internal/application/service/tenant.go index 2306c9a..a726910 100644 --- a/internal/application/service/tenant.go +++ b/internal/application/service/tenant.go @@ -5,20 +5,22 @@ import ( "crypto/aes" "crypto/cipher" "crypto/rand" + "crypto/sha256" "encoding/base64" "encoding/binary" "errors" + + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" + "io" "os" "strings" "time" - - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" ) -var apiKeySecret = []byte(os.Getenv("TENANT_AES_KEY")) +var apiKeySecret = getApiKeySecret() // ListTenantsParams defines parameters for listing tenants with filtering and pagination type ListTenantsParams struct { @@ -287,3 +289,9 @@ func (r *tenantService) ExtractTenantIDFromAPIKey(apiKey string) (uint, error) { return uint(tenantID), nil } + +func getApiKeySecret() []byte { + secret := os.Getenv("TENANT_AES_KEY") + hashArray := sha256.Sum256([]byte(secret)) + return hashArray[:] +} diff --git a/internal/application/service/test_data.go b/internal/application/service/test_data.go index 49e098d..8c20a3d 100644 --- a/internal/application/service/test_data.go +++ b/internal/application/service/test_data.go @@ -8,14 +8,14 @@ import ( "os" "strconv" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/models/rerank" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/models/rerank" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // TestDataService 测试数据服务 diff --git a/internal/common/asyncq.go b/internal/common/asyncq.go index 9316f89..ae010c4 100644 --- a/internal/common/asyncq.go +++ b/internal/common/asyncq.go @@ -3,8 +3,8 @@ package common import ( "log" - "github.com/Tencent/WeKnora/internal/config" "github.com/hibiken/asynq" + "knowlege-lsxd/internal/config" ) // client is the global asyncq client instance diff --git a/internal/config/config.go b/internal/config/config.go index ed9e7fb..d74781b 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -2,9 +2,6 @@ package config import ( "fmt" - "github.com/go-viper/mapstructure/v2" - "github.com/joho/godotenv" - "github.com/spf13/viper" "log" "os" "path/filepath" @@ -12,6 +9,10 @@ import ( "strings" "sync" "time" + + "github.com/go-viper/mapstructure/v2" + "github.com/joho/godotenv" + "github.com/spf13/viper" ) // Config 应用程序总配置 @@ -193,7 +194,6 @@ func SetEnv() { envSet("DB_HOST", "127.0.0.1") envSet("DB_PORT", "5432") envSet("TZ", "Asia/Shanghai") - envSet("OTEL_EXPORTER_OTLP_ENDPOINT", "jaeger:4317") envSet("OTEL_SERVICE_NAME", "WeKnora") envSet("OTEL_TRACES_EXPORTER", "otlp") envSet("OTEL_METRICS_EXPORTER", "none") diff --git a/internal/container/cleanup.go b/internal/container/cleanup.go index fd14357..fafce37 100644 --- a/internal/container/cleanup.go +++ b/internal/container/cleanup.go @@ -5,8 +5,8 @@ import ( "log" "sync" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // ResourceCleaner is a resource cleaner that can be used to clean up resources diff --git a/internal/container/container.go b/internal/container/container.go index 7453034..e4ab614 100644 --- a/internal/container/container.go +++ b/internal/container/container.go @@ -19,25 +19,25 @@ import ( "gorm.io/driver/postgres" "gorm.io/gorm" - "github.com/Tencent/WeKnora/internal/application/repository" - elasticsearchRepoV7 "github.com/Tencent/WeKnora/internal/application/repository/retriever/elasticsearch/v7" - elasticsearchRepoV8 "github.com/Tencent/WeKnora/internal/application/repository/retriever/elasticsearch/v8" - postgresRepo "github.com/Tencent/WeKnora/internal/application/repository/retriever/postgres" - "github.com/Tencent/WeKnora/internal/application/service" - chatpipline "github.com/Tencent/WeKnora/internal/application/service/chat_pipline" - "github.com/Tencent/WeKnora/internal/application/service/file" - "github.com/Tencent/WeKnora/internal/application/service/retriever" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/handler" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/router" - "github.com/Tencent/WeKnora/internal/stream" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" - "github.com/Tencent/WeKnora/services/docreader/src/client" + "knowlege-lsxd/internal/application/repository" + elasticsearchRepoV7 "knowlege-lsxd/internal/application/repository/retriever/elasticsearch/v7" + elasticsearchRepoV8 "knowlege-lsxd/internal/application/repository/retriever/elasticsearch/v8" + postgresRepo "knowlege-lsxd/internal/application/repository/retriever/postgres" + "knowlege-lsxd/internal/application/service" + chatpipline "knowlege-lsxd/internal/application/service/chat_pipline" + "knowlege-lsxd/internal/application/service/file" + "knowlege-lsxd/internal/application/service/retriever" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/handler" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/router" + "knowlege-lsxd/internal/stream" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" + "knowlege-lsxd/services/docreader/src/client" ) // BuildContainer constructs the dependency injection container diff --git a/internal/handler/chunk.go b/internal/handler/chunk.go index 0dbdb1c..79253b7 100644 --- a/internal/handler/chunk.go +++ b/internal/handler/chunk.go @@ -3,12 +3,12 @@ package handler import ( "net/http" - "github.com/Tencent/WeKnora/internal/application/service" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/application/service" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // ChunkHandler defines HTTP handlers for chunk operations diff --git a/internal/handler/evaluation.go b/internal/handler/evaluation.go index ac62092..f41b335 100644 --- a/internal/handler/evaluation.go +++ b/internal/handler/evaluation.go @@ -3,11 +3,11 @@ package handler import ( "net/http" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // EvaluationHandler handles evaluation related HTTP requests diff --git a/internal/handler/initialization.go b/internal/handler/initialization.go index c010d25..03a3798 100644 --- a/internal/handler/initialization.go +++ b/internal/handler/initialization.go @@ -13,18 +13,18 @@ import ( "strconv" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" - "github.com/Tencent/WeKnora/services/docreader/src/client" - "github.com/Tencent/WeKnora/services/docreader/src/proto" "github.com/gin-gonic/gin" "github.com/google/uuid" "github.com/ollama/ollama/api" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" + "knowlege-lsxd/services/docreader/src/client" + "knowlege-lsxd/services/docreader/src/proto" ) // DownloadTask 下载任务信息 diff --git a/internal/handler/knowledge.go b/internal/handler/knowledge.go index dac6ffe..89c5e35 100644 --- a/internal/handler/knowledge.go +++ b/internal/handler/knowledge.go @@ -7,11 +7,11 @@ import ( "net/http" "strconv" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // KnowledgeHandler processes HTTP requests related to knowledge resources diff --git a/internal/handler/knowledgebase.go b/internal/handler/knowledgebase.go index c72b6cb..f293d6c 100644 --- a/internal/handler/knowledgebase.go +++ b/internal/handler/knowledgebase.go @@ -3,11 +3,11 @@ package handler import ( "net/http" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // KnowledgeBaseHandler defines the HTTP handler for knowledge base operations diff --git a/internal/handler/message.go b/internal/handler/message.go index f84a18c..37b73eb 100644 --- a/internal/handler/message.go +++ b/internal/handler/message.go @@ -7,9 +7,9 @@ import ( "github.com/gin-gonic/gin" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types/interfaces" ) // MessageHandler handles HTTP requests related to messages within chat sessions diff --git a/internal/handler/model.go b/internal/handler/model.go index 7be800a..4c1f7ca 100644 --- a/internal/handler/model.go +++ b/internal/handler/model.go @@ -3,12 +3,12 @@ package handler import ( "net/http" - "github.com/Tencent/WeKnora/internal/application/service" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/application/service" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // ModelHandler handles HTTP requests for model-related operations diff --git a/internal/handler/session.go b/internal/handler/session.go index 7149e19..f62d234 100644 --- a/internal/handler/session.go +++ b/internal/handler/session.go @@ -7,13 +7,13 @@ import ( "net/http" "time" - "github.com/Tencent/WeKnora/internal/application/service" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/application/service" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // SessionHandler handles all HTTP requests related to conversation sessions diff --git a/internal/handler/tenant.go b/internal/handler/tenant.go index c862894..9a0da6e 100644 --- a/internal/handler/tenant.go +++ b/internal/handler/tenant.go @@ -6,10 +6,10 @@ import ( "github.com/gin-gonic/gin" - "github.com/Tencent/WeKnora/internal/errors" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/errors" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // TenantHandler implements HTTP request handlers for tenant management diff --git a/internal/handler/test_data.go b/internal/handler/test_data.go index 0e92c65..da60b7b 100644 --- a/internal/handler/test_data.go +++ b/internal/handler/test_data.go @@ -5,10 +5,10 @@ import ( "github.com/gin-gonic/gin" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // TestDataHandler handles HTTP requests related to test data operations diff --git a/internal/logger/logger.go b/internal/logger/logger.go index fd9ef9d..55228e5 100644 --- a/internal/logger/logger.go +++ b/internal/logger/logger.go @@ -8,8 +8,8 @@ import ( "sort" "strings" - "github.com/Tencent/WeKnora/internal/types" "github.com/sirupsen/logrus" + "knowlege-lsxd/internal/types" ) // LogLevel 日志级别类型 diff --git a/internal/middleware/auth.go b/internal/middleware/auth.go index 7f81ef5..caa4820 100644 --- a/internal/middleware/auth.go +++ b/internal/middleware/auth.go @@ -8,10 +8,10 @@ import ( "slices" "strings" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/gin-gonic/gin" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // 无需认证的API列表 diff --git a/internal/middleware/error_handler.go b/internal/middleware/error_handler.go index 2379e6a..11f5f64 100644 --- a/internal/middleware/error_handler.go +++ b/internal/middleware/error_handler.go @@ -5,7 +5,7 @@ import ( "github.com/gin-gonic/gin" - "github.com/Tencent/WeKnora/internal/errors" + "knowlege-lsxd/internal/errors" ) // ErrorHandler 是一个处理应用错误的中间件 diff --git a/internal/middleware/logger.go b/internal/middleware/logger.go index 346b91e..54370b7 100644 --- a/internal/middleware/logger.go +++ b/internal/middleware/logger.go @@ -4,10 +4,10 @@ import ( "context" "time" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/types" "github.com/gin-gonic/gin" "github.com/google/uuid" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/types" ) // RequestID middleware adds a unique request ID to the context diff --git a/internal/middleware/trace.go b/internal/middleware/trace.go index e7febfc..d74513f 100644 --- a/internal/middleware/trace.go +++ b/internal/middleware/trace.go @@ -10,8 +10,8 @@ import ( "go.opentelemetry.io/otel/attribute" "go.opentelemetry.io/otel/codes" - "github.com/Tencent/WeKnora/internal/tracing" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/tracing" + "knowlege-lsxd/internal/types" ) // Custom ResponseWriter to capture response content diff --git a/internal/models/chat/chat.go b/internal/models/chat/chat.go index 04d300d..05ff640 100644 --- a/internal/models/chat/chat.go +++ b/internal/models/chat/chat.go @@ -5,9 +5,9 @@ import ( "fmt" "strings" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/runtime" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/runtime" + "knowlege-lsxd/internal/types" ) // ChatOptions 聊天选项 diff --git a/internal/models/chat/ollama.go b/internal/models/chat/ollama.go index 4371f29..594cac3 100644 --- a/internal/models/chat/ollama.go +++ b/internal/models/chat/ollama.go @@ -4,10 +4,10 @@ import ( "context" "fmt" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/types" ollamaapi "github.com/ollama/ollama/api" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/types" ) // OllamaChat 实现了基于 Ollama 的聊天 diff --git a/internal/models/chat/remote_api.go b/internal/models/chat/remote_api.go index 4260f79..2dd8ffb 100644 --- a/internal/models/chat/remote_api.go +++ b/internal/models/chat/remote_api.go @@ -4,8 +4,8 @@ import ( "context" "fmt" - "github.com/Tencent/WeKnora/internal/types" "github.com/sashabaranov/go-openai" + "knowlege-lsxd/internal/types" ) // RemoteAPIChat 实现了基于的聊天 diff --git a/internal/models/embedding/batch.go b/internal/models/embedding/batch.go index 2849173..527b3c1 100644 --- a/internal/models/embedding/batch.go +++ b/internal/models/embedding/batch.go @@ -4,8 +4,8 @@ import ( "context" "sync" - "github.com/Tencent/WeKnora/internal/models/utils" "github.com/panjf2000/ants/v2" + "knowlege-lsxd/internal/models/utils" ) type batchEmbedder struct { diff --git a/internal/models/embedding/embedder.go b/internal/models/embedding/embedder.go index ece81c0..9f7b55a 100644 --- a/internal/models/embedding/embedder.go +++ b/internal/models/embedding/embedder.go @@ -5,9 +5,9 @@ import ( "fmt" "strings" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" - "github.com/Tencent/WeKnora/internal/runtime" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/models/utils/ollama" + "knowlege-lsxd/internal/runtime" + "knowlege-lsxd/internal/types" ) // Embedder defines the interface for text vectorization diff --git a/internal/models/embedding/ollama.go b/internal/models/embedding/ollama.go index 12e5196..ea545c5 100644 --- a/internal/models/embedding/ollama.go +++ b/internal/models/embedding/ollama.go @@ -5,9 +5,9 @@ import ( "fmt" "time" - "github.com/Tencent/WeKnora/internal/logger" - "github.com/Tencent/WeKnora/internal/models/utils/ollama" ollamaapi "github.com/ollama/ollama/api" + "knowlege-lsxd/internal/logger" + "knowlege-lsxd/internal/models/utils/ollama" ) // OllamaEmbedder implements text vectorization functionality using Ollama diff --git a/internal/models/embedding/openai.go b/internal/models/embedding/openai.go index ef987d4..4e529d1 100644 --- a/internal/models/embedding/openai.go +++ b/internal/models/embedding/openai.go @@ -9,7 +9,7 @@ import ( "net/http" "time" - "github.com/Tencent/WeKnora/internal/logger" + "knowlege-lsxd/internal/logger" ) // OpenAIEmbedder implements text vectorization functionality using OpenAI API diff --git a/internal/models/rerank/remote_api.go b/internal/models/rerank/remote_api.go index f0ad8ea..7ae6760 100644 --- a/internal/models/rerank/remote_api.go +++ b/internal/models/rerank/remote_api.go @@ -8,7 +8,7 @@ import ( "io" "net/http" - "github.com/Tencent/WeKnora/internal/logger" + "knowlege-lsxd/internal/logger" ) // OpenAIReranker implements a reranking system based on OpenAI models diff --git a/internal/models/rerank/reranker.go b/internal/models/rerank/reranker.go index f62c66b..82404a1 100644 --- a/internal/models/rerank/reranker.go +++ b/internal/models/rerank/reranker.go @@ -5,7 +5,7 @@ import ( "fmt" "strings" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // Reranker defines the interface for document reranking diff --git a/internal/models/utils/ollama/ollama.go b/internal/models/utils/ollama/ollama.go index 76b88ce..ff824ff 100644 --- a/internal/models/utils/ollama/ollama.go +++ b/internal/models/utils/ollama/ollama.go @@ -9,8 +9,8 @@ import ( "strings" "sync" - "github.com/Tencent/WeKnora/internal/logger" "github.com/ollama/ollama/api" + "knowlege-lsxd/internal/logger" ) // OllamaService manages Ollama service diff --git a/internal/router/router.go b/internal/router/router.go index 680c739..8fddce8 100644 --- a/internal/router/router.go +++ b/internal/router/router.go @@ -7,10 +7,10 @@ import ( "github.com/gin-gonic/gin" "go.uber.org/dig" - "github.com/Tencent/WeKnora/internal/config" - "github.com/Tencent/WeKnora/internal/handler" - "github.com/Tencent/WeKnora/internal/middleware" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/config" + "knowlege-lsxd/internal/handler" + "knowlege-lsxd/internal/middleware" + "knowlege-lsxd/internal/types/interfaces" ) // RouterParams 路由参数 @@ -53,7 +53,7 @@ func NewRouter(params RouterParams) *gin.Engine { r.Use(middleware.Auth(params.TenantService, params.Config)) // 添加OpenTelemetry追踪中间件 - r.Use(middleware.TracingMiddleware()) + //r.Use(middleware.TracingMiddleware()) // 健康检查 r.GET("/health", func(c *gin.Context) { diff --git a/internal/stream/factory.go b/internal/stream/factory.go index 7205441..1bdbf2d 100644 --- a/internal/stream/factory.go +++ b/internal/stream/factory.go @@ -5,7 +5,7 @@ import ( "strconv" "time" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/types/interfaces" ) // 流管理器类型 diff --git a/internal/stream/memory_manager.go b/internal/stream/memory_manager.go index 3210eea..2cc0240 100644 --- a/internal/stream/memory_manager.go +++ b/internal/stream/memory_manager.go @@ -5,8 +5,8 @@ import ( "sync" "time" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // 内存流信息 diff --git a/internal/stream/redis_manager.go b/internal/stream/redis_manager.go index ec4e138..e993e7a 100644 --- a/internal/stream/redis_manager.go +++ b/internal/stream/redis_manager.go @@ -6,9 +6,9 @@ import ( "fmt" "time" - "github.com/Tencent/WeKnora/internal/types" - "github.com/Tencent/WeKnora/internal/types/interfaces" "github.com/redis/go-redis/v9" + "knowlege-lsxd/internal/types" + "knowlege-lsxd/internal/types/interfaces" ) // redisStreamInfo Redis存储的流信息 diff --git a/internal/types/interfaces/chunk.go b/internal/types/interfaces/chunk.go index 81ea87b..9a43783 100644 --- a/internal/types/interfaces/chunk.go +++ b/internal/types/interfaces/chunk.go @@ -3,7 +3,7 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // ChunkRepository defines the interface for chunk repository operations diff --git a/internal/types/interfaces/evaluation.go b/internal/types/interfaces/evaluation.go index db89a79..e4fd4e4 100644 --- a/internal/types/interfaces/evaluation.go +++ b/internal/types/interfaces/evaluation.go @@ -3,7 +3,7 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // EvaluationService defines operations for evaluation tasks diff --git a/internal/types/interfaces/knowledge.go b/internal/types/interfaces/knowledge.go index 621385b..78e0eb8 100644 --- a/internal/types/interfaces/knowledge.go +++ b/internal/types/interfaces/knowledge.go @@ -5,7 +5,7 @@ import ( "io" "mime/multipart" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // KnowledgeService defines the interface for knowledge services. diff --git a/internal/types/interfaces/knowledgebase.go b/internal/types/interfaces/knowledgebase.go index 38bcf34..73d823e 100644 --- a/internal/types/interfaces/knowledgebase.go +++ b/internal/types/interfaces/knowledgebase.go @@ -7,7 +7,7 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // KnowledgeBaseService defines the knowledge base service interface diff --git a/internal/types/interfaces/message.go b/internal/types/interfaces/message.go index ffb1777..cef51e6 100644 --- a/internal/types/interfaces/message.go +++ b/internal/types/interfaces/message.go @@ -4,7 +4,7 @@ import ( "context" "time" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // MessageService defines the message service interface diff --git a/internal/types/interfaces/model.go b/internal/types/interfaces/model.go index 7742222..115a6f1 100644 --- a/internal/types/interfaces/model.go +++ b/internal/types/interfaces/model.go @@ -3,10 +3,10 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/models/chat" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/models/rerank" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/models/chat" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/models/rerank" + "knowlege-lsxd/internal/types" ) // ModelService defines the model service interface diff --git a/internal/types/interfaces/resource.go b/internal/types/interfaces/resource.go index 07c6cd8..a9333f9 100644 --- a/internal/types/interfaces/resource.go +++ b/internal/types/interfaces/resource.go @@ -3,7 +3,7 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // ResourceCleaner defines the resource cleaner interface diff --git a/internal/types/interfaces/retriever.go b/internal/types/interfaces/retriever.go index 7969ab7..7e7fa24 100644 --- a/internal/types/interfaces/retriever.go +++ b/internal/types/interfaces/retriever.go @@ -3,8 +3,8 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/models/embedding" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/models/embedding" + "knowlege-lsxd/internal/types" ) // RetrieveEngine defines the retrieve engine interface diff --git a/internal/types/interfaces/session.go b/internal/types/interfaces/session.go index 1f5c5ab..576708a 100644 --- a/internal/types/interfaces/session.go +++ b/internal/types/interfaces/session.go @@ -3,7 +3,7 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // SessionService defines the session service interface diff --git a/internal/types/interfaces/stream_manager.go b/internal/types/interfaces/stream_manager.go index d6a26ca..751324c 100644 --- a/internal/types/interfaces/stream_manager.go +++ b/internal/types/interfaces/stream_manager.go @@ -4,7 +4,7 @@ import ( "context" "time" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // StreamInfo stream information diff --git a/internal/types/interfaces/tenant.go b/internal/types/interfaces/tenant.go index df5839e..20811ee 100644 --- a/internal/types/interfaces/tenant.go +++ b/internal/types/interfaces/tenant.go @@ -3,7 +3,7 @@ package interfaces import ( "context" - "github.com/Tencent/WeKnora/internal/types" + "knowlege-lsxd/internal/types" ) // TenantService defines the tenant service interface diff --git a/scripts/start_all.sh b/scripts/start_all.sh index 070ce12..8f29dac 100755 --- a/scripts/start_all.sh +++ b/scripts/start_all.sh @@ -642,7 +642,6 @@ else log_success "所有服务启动完成，可通过以下地址访问:" echo -e "${GREEN} - 前端界面: http://localhost${NC}" echo -e "${GREEN} - API接口: http://localhost:8080${NC}" - echo -e "${GREEN} - Jaeger链路追踪: http://localhost:16686${NC}" else log_error "部分服务启动失败，请检查日志并修复问题" fi @@ -653,7 +652,6 @@ else log_success "Docker容器启动完成，可通过以下地址访问:" echo -e "${GREEN} - 前端界面: http://localhost${NC}" echo -e "${GREEN} - API接口: http://localhost:8080${NC}" - echo -e "${GREEN} - Jaeger链路追踪: http://localhost:16686${NC}" fi fi diff --git a/services/docreader/src/client/client.go b/services/docreader/src/client/client.go index ee74246..e69a1ea 100644 --- a/services/docreader/src/client/client.go +++ b/services/docreader/src/client/client.go @@ -6,10 +6,10 @@ import ( "os" "time" - "github.com/Tencent/WeKnora/services/docreader/src/proto" "google.golang.org/grpc" "google.golang.org/grpc/credentials/insecure" "google.golang.org/grpc/resolver" + "knowlege-lsxd/services/docreader/src/proto" ) const ( diff --git a/services/docreader/src/client/client_test.go b/services/docreader/src/client/client_test.go index 5551048..3476dcc 100644 --- a/services/docreader/src/client/client_test.go +++ b/services/docreader/src/client/client_test.go @@ -7,7 +7,7 @@ import ( "testing" "time" - "github.com/Tencent/WeKnora/services/docreader/src/proto" + "knowlege-lsxd/services/docreader/src/proto" ) func init() { diff --git a/services/docreader/src/proto/docreader.pb.go b/services/docreader/src/proto/docreader.pb.go index 95199af..9c29c8f 100644 --- a/services/docreader/src/proto/docreader.pb.go +++ b/services/docreader/src/proto/docreader.pb.go @@ -1,7 +1,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: // protoc-gen-go v1.36.6 -// protoc v5.29.3 +// protoc v4.25.3 // source: docreader.proto package proto @@ -747,7 +747,7 @@ const file_docreader_proto_rawDesc = "" + "\x05MINIO\x10\x022\x9f\x01\n" + "\tDocReader\x12I\n" + "\fReadFromFile\x12\x1e.docreader.ReadFromFileRequest\x1a\x17.docreader.ReadResponse\"\x00\x12G\n" + - "\vReadFromURL\x12\x1d.docreader.ReadFromURLRequest\x1a\x17.docreader.ReadResponse\"\x00B5Z3github.com/Tencent/WeKnora/internal/docreader/protob\x06proto3" + "\vReadFromURL\x12\x1d.docreader.ReadFromURLRequest\x1a\x17.docreader.ReadResponse\"\x00B(Z&knowlege-lsxd/internal/docreader/protob\x06proto3" var ( file_docreader_proto_rawDescOnce sync.Once diff --git a/services/docreader/src/proto/docreader.proto b/services/docreader/src/proto/docreader.proto index 44ff4ac..45e1b74 100644 --- a/services/docreader/src/proto/docreader.proto +++ b/services/docreader/src/proto/docreader.proto @@ -2,7 +2,7 @@ syntax = "proto3"; package docreader; -option go_package = "github.com/Tencent/WeKnora/internal/docreader/proto"; +option go_package = "knowlege-lsxd/internal/docreader/proto"; // 文档读取服务 service DocReader { diff --git a/services/docreader/src/proto/docreader_grpc.pb.go b/services/docreader/src/proto/docreader_grpc.pb.go index 79bd7ef..08ae678 100644 --- a/services/docreader/src/proto/docreader_grpc.pb.go +++ b/services/docreader/src/proto/docreader_grpc.pb.go @@ -1,7 +1,7 @@ // Code generated by protoc-gen-go-grpc. DO NOT EDIT. // versions: // - protoc-gen-go-grpc v1.5.1 -// - protoc v5.29.3 +// - protoc v4.25.3 // source: docreader.proto package proto diff --git a/services/docreader/src/proto/docreader_pb2_grpc.py b/services/docreader/src/proto/docreader_pb2_grpc.py index 1b4eaf7..6410d60 100644 --- a/services/docreader/src/proto/docreader_pb2_grpc.py +++ b/services/docreader/src/proto/docreader_pb2_grpc.py @@ -3,7 +3,7 @@ import grpc import warnings -from . import docreader_pb2 as docreader__pb2 +from . from . import docreader_pb2 as docreader__pb2 GRPC_GENERATED_VERSION = '1.74.0' GRPC_VERSION = grpc.__version__