From 131213d2499b5b5a8e93e2ec7de39f02dd98143f Mon Sep 17 00:00:00 2001 From: Palaniappan R Date: Tue, 29 Oct 2024 00:54:02 +0530 Subject: [PATCH] fix: add opensta to general retriever Signed-off-by: Palaniappan R --- backend/src/agents/retriever_tools.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/backend/src/agents/retriever_tools.py b/backend/src/agents/retriever_tools.py index fc1d741..18f21b9 100644 --- a/backend/src/agents/retriever_tools.py +++ b/backend/src/agents/retriever_tools.py @@ -56,8 +56,9 @@ def initialize( "./data/markdown/gh_discussions", "./data/markdown/manpages/man1", "./data/markdown/manpages/man2", + "./data/markdown/OpenSTA_docs", ], - other_docs_path=["./data/pdf/OR_publications"], + other_docs_path=["./data/pdf"], weights=[0.6, 0.2, 0.2], contextual_rerank=True, search_k=search_k, @@ -75,6 +76,7 @@ def initialize( "./data/markdown/OR_docs/installation", "./data/markdown/gh_discussions/Build", "./data/markdown/gh_discussions/Installation", + "./data/markdown/OpenSTA_docs/", ], weights=[0.6, 0.2, 0.2], contextual_rerank=True, @@ -153,7 +155,7 @@ def initialize( @tool def retrieve_general(query: str) -> Tuple[str, list[str], list[str]]: """ - Retrieve comprehensive and detailed information pertaining to the OpenROAD project and OpenROAD-Flow-Scripts.\ + Retrieve comprehensive and detailed information pertaining to the OpenROAD project, OpenROAD-Flow-Scripts and OpenSTA.\ This includes, but is not limited to, general information, specific functionalities, usage guidelines,\ troubleshooting steps, and best practices. The tool is designed to assist users by providing clear, accurate,\ and relevant information that enhances their understanding and efficient use of OpenROAD and OpenROAD-Flow-Scripts.\ @@ -197,7 +199,7 @@ def retrieve_cmds(query: str) -> Tuple[str, list[str], list[str]]: @tool def retrieve_install(query: str) -> Tuple[str, list[str], list[str]]: """ - Retrieve comprehensive and detailed information pertaining to the installaion of OpenROAD project and OpenROAD-Flow-Scripts.\ + Retrieve comprehensive and detailed information pertaining to the installaion of OpenROAD, OpenROAD-Flow-Scripts and OpenSTA.\ This includes, but is not limited to, various dependencies, system requirements, installation methods such as,\ - Building from source\ - Using Docker\