github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-07 19:34:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

style:fmt

Browse Source
This commit is contained in:
aries_ckt 2023-08-10 14:38:50 +08:00
parent 84fbebda4c
commit a7ec3400fe
4 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/configs/config.py
View File

@ -165,7 +165,9 @@ class Config(metaclass=Singleton):
self.KNOWLEDGE_CHUNK_SIZE = int(os.getenv("KNOWLEDGE_CHUNK_SIZE", 100)) self.KNOWLEDGE_CHUNK_SIZE = int(os.getenv("KNOWLEDGE_CHUNK_SIZE", 100))
self.KNOWLEDGE_CHUNK_OVERLAP = int(os.getenv("KNOWLEDGE_CHUNK_OVERLAP", 100)) self.KNOWLEDGE_CHUNK_OVERLAP = int(os.getenv("KNOWLEDGE_CHUNK_OVERLAP", 100))
self.KNOWLEDGE_SEARCH_TOP_SIZE = int(os.getenv("KNOWLEDGE_SEARCH_TOP_SIZE", 5)) self.KNOWLEDGE_SEARCH_TOP_SIZE = int(os.getenv("KNOWLEDGE_SEARCH_TOP_SIZE", 5))
self.KNOWLEDGE_SEARCH_MAX_TOKEN = int(os.getenv("KNOWLEDGE_SEARCH_MAX_TOKEN", 2000)) self.KNOWLEDGE_SEARCH_MAX_TOKEN = int(
os.getenv("KNOWLEDGE_SEARCH_MAX_TOKEN", 2000)
)
### SUMMARY_CONFIG Configuration ### SUMMARY_CONFIG Configuration
self.SUMMARY_CONFIG = os.getenv("SUMMARY_CONFIG", "FAST") self.SUMMARY_CONFIG = os.getenv("SUMMARY_CONFIG", "FAST")

2
pilot/scene/chat_knowledge/v1/prompt.py
View File

@ -12,7 +12,7 @@ from pilot.scene.chat_normal.out_parser import NormalChatOutputParser
CFG = Config() CFG = Config()
PROMPT_SCENE_DEFINE = """A chat between a curious user and an artificial intelligence assistant, who very familiar with database related knowledge. PROMPT_SCENE_DEFINE = """A chat between a curious user and an artificial intelligence assistant, who very familiar with database related knowledge.
The assistant gives helpful, detailed, professional and polite answers to the user's questions. """ The assistant gives helpful, detailed, professional and polite answers to the user's questions. """
_DEFAULT_TEMPLATE_ZH = """ 基于以下已知的信息, 专业、简要的回答用户的问题, _DEFAULT_TEMPLATE_ZH = """ 基于以下已知的信息, 专业、简要的回答用户的问题,

2
pilot/server/knowledge/service.py
View File

@ -188,7 +188,7 @@ class KnowledgeService:
text_splitter = SpacyTextSplitter( text_splitter = SpacyTextSplitter(
pipeline="zh_core_web_sm", pipeline="zh_core_web_sm",
chunk_size=chunk_size, chunk_size=chunk_size,
chunk_overlap=chunk_overlap chunk_overlap=chunk_overlap,
) )
except Exception: except Exception:
text_splitter = RecursiveCharacterTextSplitter( text_splitter = RecursiveCharacterTextSplitter(

4
pilot/vector_store/chroma_store.py
View File

@ -26,10 +26,10 @@ class ChromaStore(VectorStoreBase):
persist_directory=self.persist_dir, persist_directory=self.persist_dir,
embedding_function=self.embeddings, embedding_function=self.embeddings,
client_settings=chroma_settings, client_settings=chroma_settings,
collection_metadata=collection_metadata collection_metadata=collection_metadata,
) )
def similar_search(self, text, topk, **kwargs: Any) -> None: def similar_search(self, text, topk, **kwargs: Any) -> None:
logger.info("ChromaStore similar search") logger.info("ChromaStore similar search")
return self.vector_store_client.similarity_search(text, topk) return self.vector_store_client.similarity_search(text, topk)