Skip to content

Commit

Permalink
Fixed pylint score
Browse files Browse the repository at this point in the history
  • Loading branch information
antoninoLorenzo committed Sep 1, 2024
1 parent 0fde5d0 commit e7a4b81
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions src/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,16 +85,16 @@ class APISettings(BaseSettings):
)

initialize_knowledge(rag_settings.DOCS_BASE_PATH, store)
available = ''
available_documents = ''
for name, coll in store.collections.items():
topics = ", ".join([topic.name for topic in coll.topics])
available += f"- '{name}': {topics}\n"
doc_topics = ", ".join([topic.name for topic in coll.topics])
available_documents += f"- '{name}': {doc_topics}\n"


@TR.register(
description=f"""Search documents in a Retrieval Augmented Generation Vector Database.
description=f"""Search documents in the RAG Vector Database.
Available collections are:
{available}
{available_documents}
"""
)
def search_rag(rag_query: str, collection: str) -> str:
Expand Down Expand Up @@ -128,6 +128,7 @@ def search_rag(rag_query: str, collection: str) -> str:

@app.get('/')
def ping():
""""""
return ''


Expand Down

0 comments on commit e7a4b81

Please sign in to comment.