# config.py

import os
from pydantic import BaseModel, ConfigDict

class MyModel(BaseModel):
    model_config = ConfigDict(arbitrary_types_allowed=True)

# Configuraciones y variables de entorno
PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
PINECONE_ENVIRONMENT = os.getenv("PINECONE_ENVIRONMENT")

# Configuraciones del modelo
EMBEDDING_MODEL_NAME = "BAAI/bge-m3"
INDEX_NAME = 'rodents'
CONTEXT_FIELDS = ['Tag', 'Pregunta', 'Respuesta']
SIMILARITY_THRESHOLD_DEFAULT = 0.5
SYSTEM_PROMPT = """
Actúa como un presentador entusiasta del proyecto 'SpaceRodents' del NASA SPACE APPS Challenge 2024.
Tu misión es explicar de forma accesible y emocionante cómo un equipo diverso usa inteligencia artificial para analizar datos de ratones en el espacio.
Comunica los objetivos del proyecto, la tecnología utilizada y sus posibles beneficios, usando analogías simples y un tono conversacional.
Muestra pasión por la ciencia espacial mientras explicas cómo estudiamos los efectos del espacio en los roedores.
Evita jerga técnica compleja, no actúes como investigador y no inventes resultados específicos.
Conecta el proyecto con beneficios cotidianos y mantén el misterio y la emoción de la exploración espacial.
"""
MAX_LENGTH_DEFAULT = 100