/
igormayer
/
langgraph_sql_agent_web
Обзор
Документация
Войти
/
igormayer
/
langgraph_sql_agent_web
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
backend/sqlagent/factory.py
41 строка
1 KB
Igor Mayer
fixes for running in docker
08 фев 2026, 21:16
08 фев 2026, 21:16
988621e
Код
Авторство
О чём код?
from langchain_community.agent_toolkits import SQLDatabaseToolkit from langchain_community.utilities import SQLDatabase from langchain_ollama import ChatOllama from langchain_openai import ChatOpenAI # function is used in graph nodes def get_toolkit(config: dict): # get params from config if config is not None: conf = config.get('configurable', {}) db_uri = f'{conf.get("db_dialect", "")}://{conf.get("db_user", "")}:{conf.get("db_password", "")}@{conf.get("db_host", "")}:{conf.get("db_port", "")}/{conf.get("db_name", "")}' model_name = conf.get('model_name', "") model_host_url = conf.get('model_host_url', "") model_host_type = conf.get('model_host_type', "") model_api_key = conf.get('model_api_key', "") # toolkit init db = SQLDatabase.from_uri(db_uri) match model_host_type: case "OLLAMA": llm = ChatOllama( base_url=model_host_url, model=model_name, temperature=0.0 ) case "OPENAI API": llm = ChatOpenAI( base_url=model_host_url, model=model_name, api_key=model_api_key, temperature=0.0, max_tokens=None, timeout=None, max_retries=2 ) toolkit = SQLDatabaseToolkit(db=db, llm=llm) tools_map = {t.name: t for t in toolkit.get_tools()} return tools_map, llm