alihan/KG_RAG
1
0
Fork 0
mirror of https://github.com/BaranziniLab/KG_RAG.git synced 2024-06-08 14:12:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix(GPT_API_TYPE): fix a bug that occurs when setting GPT_API_TYPE equal to 'openai'

Browse Source 
There is no CHAT_DEPLOYMENT_ID parameter in openai's API. Adding it will cause an error.
This commit is contained in:
Jarlor Zhang
2024-04-06 18:12:44 +08:00
parent 42fe2418de
commit 9a6af6b93f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kg_rag/rag_based_generation/GPT/text_generation.py
View File

@@ -30,8 +30,8 @@ SENTENCE_EMBEDDING_MODEL_FOR_NODE_RETRIEVAL = config_data["SENTENCE_EMBEDDING_MO
SENTENCE_EMBEDDING_MODEL_FOR_CONTEXT_RETRIEVAL = config_data["SENTENCE_EMBEDDING_MODEL_FOR_CONTEXT_RETRIEVAL"]
TEMPERATURE = config_data["LLM_TEMPERATURE"]
CHAT_DEPLOYMENT_ID = CHAT_MODEL_ID
# can not specify CHAT_DEPLOYMENT_ID if GPT_API_TYPE is 'openai'.Otherwise it will be not work
CHAT_DEPLOYMENT_ID = CHAT_MODEL_ID if openai.api_type == "azure" else None
vectorstore = load_chroma(VECTOR_DB_PATH, SENTENCE_EMBEDDING_MODEL_FOR_NODE_RETRIEVAL)