SkywardAI / rebel

This is the UI for SkywardAI APIs aggregator, written in native javascript.
https://frontend-js-fqn1.onrender.com
0 stars 0 forks source link

[bug v0.1.8]RAG can't get correct collection_name of vectordb #55

Closed Aisuko closed 1 week ago

Aisuko commented 1 month ago

Please check the parameters of API chat When I use UI in RAG mode

milvus-standalone  | [2024/07/27 00:03:25.329 +00:00] [WARN] [proxy/impl.go:2610] ["Search failed to WaitToFinish"] [traceID=0b63c55fcace5b8b1f11f0b05dfd483e] [role=proxy] [db=default] [collection=collection_name] [partitions="[]"] [dsl=] [len(PlaceholderGroup)=1548] [OutputFields="[answer]"] [search_params="[{\"key\":\"topk\",\"value\":\"1\"},{\"key\":\"params\",\"value\":\"{}\"},{\"key\":\"round_decimal\",\"value\":\"-1\"},{\"key\":\"ignore_growing\",\"value\":\"False\"},{\"key\":\"metric_type\",\"value\":\"COSINE\"}]"] [guarantee_timestamp=1] [nq=0] [error="collection not found[database=default][collection=collection_name]"]
kirin              | RPC error: [search], <MilvusException: (code=100, message=collection not found[database=default][collection=collection_name])>, <Time:{'RPC start': '2024-07-27 00:03:25.327647', 'RPC error': '2024-07-27 00:03:25.329498'}>
kirin              | Failed to search collection: collection_name
kirin              | 2024-07-27 00:03:25.329 | ERROR    | src.repository.vector_database:search:79 - <MilvusException: (code=100, message=collection not found[database=default][collection=collection_name])>

When I use RAG mode with swagger

curl -X 'POST' \
  'http://localhost:8000/api/chat' \
  -H 'accept: application/json' \
  -H 'Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VybmFtZSI6InRlc3QiLCJlbWFpbCI6InRlc3RAZ21haWwuY29tIiwiZXhwIjoxNzIyNTM1NjkwLCJzdWIiOiJZT1VSLUpXVC1TVUJKRUNUIn0.IkEw6ySG1tTR8FjzGjkyJTN8PwnZ21BUNpqCtT3-DQ0' \
  -H 'Content-Type: application/json' \
  -d '{
  "sessionUuid": "90fda911-1fa8-4164-917b-2e9252751bf4",
  "message": "hi",
  "temperature": 0,
  "topK": 0,
  "topP": 0,
  "nPredict": 0,
  "collectionName": "aisuko/squad01"
}'

Logs are below

kirin              | 2024-07-27 00:09:34.917 | INFO     | src.repository.vector_database:search:72 - Vector Database --- Result: [[{'id': 451421852620293337, 'distance': 0.1441650539636612, 'entity': {'answer': 'Jonathan Stewart'}}]]