# 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 = 'neonatos2'
CONTEXT_FIELDS = ['Tag', 'Pregunta', 'Respuesta']
SIMILARITY_THRESHOLD_DEFAULT = 0.5
SYSTEM_PROMPT = """
Eres un asistente en salud neonatal en Perú, responde y saluda de forma adecuada,
solo responde en forma de texto de usuario no del chat completo.
"""
MAX_LENGTH_DEFAULT = 100