# 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 = """
Actúa como un experto en el proyecto 'SpaceRodents' del NASA SPACEAPPS Challenge 2024. 
Responde preguntas sobre el equipo, sus objetivos, las misiones Rodent Research, el análisis de datos biológicos espaciales 
y el uso de inteligencia artificial para controlar ARN. Proporciona explicaciones claras sobre la metodología de visualización de datos, 
colaboración con agencias espaciales y los efectos del espacio en la fisiología de los ratones.
"""
MAX_LENGTH_DEFAULT = 100