LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 16:43:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update application files, fix LLM models, and create new retriever class

Browse Source
This commit is contained in:
Alex
2024-04-09 16:38:42 +01:00
parent 1e26943c3e
commit 19494685ba
6 changed files with 67 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
application/api/answer/routes.py
View File

@@ -205,7 +205,12 @@ def stream():
else:
source = {}
retriever = RetrieverCreator.create_retriever("classic", question=question,
if source["active_docs"].split("/")[0] == "default" or source["active_docs"].split("/")[0] == "local":
retriever_name = "classic"
else:
retriever_name = source['active_docs']
retriever = RetrieverCreator.create_retriever(retriever_name, question=question,
source=source, chat_history=history, prompt=prompt, chunks=chunks, gpt_model=gpt_model
)
@@ -247,7 +252,12 @@ def api_answer():
else:
source = {data}
retriever = RetrieverCreator.create_retriever("classic", question=question,
if source["active_docs"].split("/")[0] == "default" or source["active_docs"].split("/")[0] == "local":
retriever_name = "classic"
else:
retriever_name = source['active_docs']
retriever = RetrieverCreator.create_retriever(retriever_name, question=question,
source=source, chat_history=history, prompt=prompt, chunks=chunks, gpt_model=gpt_model
)
source_log_docs = []
@@ -290,9 +300,14 @@ def api_search():
else:
chunks = 2
retriever = RetrieverCreator.create_retriever("classic", question=question,
source=source, chat_history=[], prompt="default", chunks=chunks, gpt_model=gpt_model
)
if source["active_docs"].split("/")[0] == "default" or source["active_docs"].split("/")[0] == "local":
retriever_name = "classic"
else:
retriever_name = source['active_docs']
retriever = RetrieverCreator.create_retriever(retriever_name, question=question,
source=source, chat_history=[], prompt="default", chunks=chunks, gpt_model=gpt_model
)
docs = retriever.search()
return docs