Spaces:
Sleeping
Sleeping
File size: 3,626 Bytes
d8a070a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import os
import yaml
from dotenv import load_dotenv
from pyprojroot import here
load_dotenv()
class LoadToolsConfig:
def __init__(self) -> None:
with open(here("configs/tools_config.yml")) as cfg:
app_config = yaml.load(cfg, Loader=yaml.FullLoader)
# Set environment variables
groq_api_key = os.getenv("GROQ_API_KEY")
tavily_api_key = os.getenv("TAVILY_API_KEY")
# Primary agent
self.primary_agent_llm = app_config["primary_agent"]["llm"]
self.primary_agent_llm_temperature = app_config["primary_agent"]["llm_temperature"]
# Internet Search config
self.tavily_search_max_results = int(
app_config["tavily_search_api"]["tavily_search_max_results"])
# Swiss Airline Policy RAG configs
self.policy_rag_llm = app_config["swiss_airline_policy_rag"]["llm"]
self.policy_rag_llm_temperature = float(
app_config["swiss_airline_policy_rag"]["llm_temperature"])
self.policy_rag_embedding_model = app_config["swiss_airline_policy_rag"]["embedding_model"]
self.policy_rag_vectordb_directory = str(here(
app_config["swiss_airline_policy_rag"]["vectordb"])) # needs to be strin for summation in chromadb backend: self._settings.require("persist_directory") + "/chroma.sqlite3"
self.policy_rag_unstructured_docs_directory = str(here(
app_config["swiss_airline_policy_rag"]["unstructured_docs"]))
self.policy_rag_k = app_config["swiss_airline_policy_rag"]["k"]
self.policy_rag_chunk_size = app_config["swiss_airline_policy_rag"]["chunk_size"]
self.policy_rag_chunk_overlap = app_config["swiss_airline_policy_rag"]["chunk_overlap"]
self.policy_rag_collection_name = app_config["swiss_airline_policy_rag"]["collection_name"]
# Stories RAG configs
self.stories_rag_llm = app_config["stories_rag"]["llm"]
self.stories_rag_llm_temperature = float(
app_config["stories_rag"]["llm_temperature"])
self.stories_rag_embedding_model = app_config["stories_rag"]["embedding_model"]
self.stories_rag_vectordb_directory = str(here(
app_config["stories_rag"]["vectordb"])) # needs to be strin for summation in chromadb backend: self._settings.require("persist_directory") + "/chroma.sqlite3"
self.stories_rag_unstructured_docs_directory = str(here(
app_config["stories_rag"]["unstructured_docs"]))
self.stories_rag_k = app_config["stories_rag"]["k"]
self.stories_rag_chunk_size = app_config["stories_rag"]["chunk_size"]
self.stories_rag_chunk_overlap = app_config["stories_rag"]["chunk_overlap"]
self.stories_rag_collection_name = app_config["stories_rag"]["collection_name"]
# Travel SQL Agent configs
self.travel_sqldb_directory = str(here(
app_config["travel_sqlagent_configs"]["travel_sqldb_dir"]))
self.travel_sqlagent_llm = app_config["travel_sqlagent_configs"]["llm"]
self.travel_sqlagent_llm_temperature = float(
app_config["travel_sqlagent_configs"]["llm_temperature"])
# Chinook SQL agent configs
self.chinook_sqldb_directory = str(here(
app_config["chinook_sqlagent_configs"]["chinook_sqldb_dir"]))
self.chinook_sqlagent_llm = app_config["chinook_sqlagent_configs"]["llm"]
self.chinook_sqlagent_llm_temperature = float(
app_config["chinook_sqlagent_configs"]["llm_temperature"])
# Graph configs
self.thread_id = str(
app_config["graph_configs"]["thread_id"])
|