SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 23:22:57 +01:00
Code Issues Projects Releases Packages Activity

Added home.py file for /chat route

Browse source
This commit is contained in:
Saurab-Shrestha 2024-01-25 17:37:00 +05:45
parent bcc3f03e25
commit 55565fd3a7
20 changed files with 295 additions and 179 deletions
Download patch file Download diff file Expand all files Collapse all files

16
private_gpt/ui/admin_ui.py
View file

@ -33,6 +33,8 @@ SOURCES_SEPARATOR = "\n\n Sources: \n"
MODES = ["Query Docs", "Search in Docs", "LLM Chat"]
# generate
@singleton
class PrivateAdminGptUi:
@inject
@ -67,11 +69,16 @@ class PrivateAdminGptUi:
if completion_gen.sources:
full_response += SOURCES_SEPARATOR
cur_sources = Source.curate_sources(completion_gen.sources)
# sources_text = "\n\n\n".join(
# f"{index}. {source.file} (page {source.page}) (page_link {source.page_link})"
# for index, source in enumerate(cur_sources, start=1)
# )
sources_text = "\n\n\n".join(
f"{index}. {source.file} (page {source.page})"
f'<a href="#" target="_blank" rel="noopener noreferrer">{index}. {source.file} (page {source.page})</a>'
for index, source in enumerate(cur_sources, start=1)
)
full_response += sources_text
print(full_response)
yield full_response
def build_history() -> list[ChatMessage]:
@ -125,11 +132,10 @@ class PrivateAdminGptUi:
)
sources = Source.curate_sources(response)
yield "\n\n\n".join(
f"{index}. **{source.file} "
f"(page {source.page})**\n "
f"{source.text}"
f"{index}. **{source.file} (page {source.page})**\n"
f" (link: [{source.page_link}]({source.page_link}))\n{source.text}"
for index, source in enumerate(sources, start=1)
)