highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-09 20:55:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix import ordering

Browse Source
This commit is contained in:
Weves
2025-05-16 14:43:33 -07:00
committed by Chris Weaver
parent 8d2e170fc4
commit 0735a98284
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/onyx/llm/utils.py
View File

@@ -419,7 +419,7 @@ def _strip_colon_from_model_name(model_name: str) -> str:
return ":".join(model_name.split(":")[:-1]) if ":" in model_name else model_name
def _find_model_obj(model_map: dict, provider: str, model_name: str) -> dict | None:
def find_model_obj(model_map: dict, provider: str, model_name: str) -> dict | None:
stripped_model_name = _strip_extra_provider_from_model_name(model_name)
model_names = [
@@ -537,7 +537,7 @@ def get_llm_max_tokens(
return GEN_AI_MAX_TOKENS
try:
model_obj = _find_model_obj(
model_obj = find_model_obj(
model_map,
model_provider,
model_name,
@@ -646,7 +646,7 @@ def get_max_input_tokens_from_llm_provider(
def model_supports_image_input(model_name: str, model_provider: str) -> bool:
model_map = get_model_map()
try:
model_obj = _find_model_obj(
model_obj = find_model_obj(
model_map,
model_provider,
model_name,

5
backend/onyx/tools/utils.py
View File

@@ -6,14 +6,15 @@ from onyx.configs.app_configs import AZURE_DALLE_API_KEY
from onyx.db.connector import check_connectors_exist
from onyx.db.document import check_docs_exist
from onyx.db.models import LLMProvider
from onyx.llm.utils import get_model_map, _find_model_obj
from onyx.llm.utils import find_model_obj
from onyx.llm.utils import get_model_map
from onyx.natural_language_processing.utils import BaseTokenizer
from onyx.tools.tool import Tool
def explicit_tool_calling_supported(model_provider: str, model_name: str) -> bool:
model_map = get_model_map()
model_obj = _find_model_obj(
model_obj = find_model_obj(
model_map=model_map,
provider=model_provider,
model_name=model_name,