highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-07 19:38:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

Built in tool cache with tool call id (#3617)

Browse Source 
* k

* improved

* k

* nit

* nit

* nit
This commit is contained in:
pablonyx 2025-01-06 17:03:52 -08:00 committed by GitHub
parent e329b63b89
commit 91903141cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
backend/onyx/tools/built_in_tools.py
View File

@ -151,7 +151,7 @@ def auto_add_search_tool_to_personas(db_session: Session) -> None:
logger.notice("Completed adding SearchTool to relevant Personas.")
_built_in_tools_cache: dict[int, Type[Tool]] | None = None
_built_in_tools_cache: dict[str, Type[Tool]] | None = None
def refresh_built_in_tools_cache(db_session: Session) -> None:
@ -173,15 +173,21 @@ def refresh_built_in_tools_cache(db_session: Session) -> None:
),
None,
)
if tool_info:
_built_in_tools_cache[tool.id] = tool_info["cls"]
if tool_info and tool.in_code_tool_id:
_built_in_tools_cache[tool.in_code_tool_id] = tool_info["cls"]
def get_built_in_tool_by_id(
tool_id: int, db_session: Session, force_refresh: bool = False
in_code_tool_id: str, db_session: Session, force_refresh: bool = False
) -> Type[Tool]:
global _built_in_tools_cache
if _built_in_tools_cache is None or force_refresh:
# If the tool is not in the cache, refresh it once
if (
_built_in_tools_cache is None
or force_refresh
or in_code_tool_id not in _built_in_tools_cache
):
refresh_built_in_tools_cache(db_session)
if _built_in_tools_cache is None:
@ -189,7 +195,9 @@ def get_built_in_tool_by_id(
"Built-in tools cache is None despite being refreshed. Should never happen."
)
if tool_id in _built_in_tools_cache:
return _built_in_tools_cache[tool_id]
else:
raise ValueError(f"No built-in tool found in the cache with ID {tool_id}")
if in_code_tool_id not in _built_in_tools_cache:
raise ValueError(
f"No built-in tool found in the cache with ID {in_code_tool_id}"
)
return _built_in_tools_cache[in_code_tool_id]

4
backend/onyx/tools/tool_constructor.py
View File

@ -148,7 +148,9 @@ def construct_tools(
for db_tool_model in persona.tools:
if db_tool_model.in_code_tool_id:
tool_cls = get_built_in_tool_by_id(db_tool_model.id, db_session)
tool_cls = get_built_in_tool_by_id(
db_tool_model.in_code_tool_id, db_session
)
# Handle Search Tool
if tool_cls.__name__ == SearchTool.__name__:

6
web/next.config.js
View File

@ -14,7 +14,11 @@ const cspHeader = `
base-uri 'self';
form-action 'self';
frame-ancestors 'none';
${process.env.NEXT_PUBLIC_CLOUD_ENABLED === "true" ? "upgrade-insecure-requests;" : ""}
${
process.env.NEXT_PUBLIC_CLOUD_ENABLED === "true"
? "upgrade-insecure-requests;"
: ""
}
`;
/** @type {import('next').NextConfig} */