File size: 1,476 Bytes
bef9637 |
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 |
# modules/chatbot/chatbot/chat_process.py
import anthropic
import logging
from typing import Dict, Generator
logger = logging.getLogger(__name__)
class ChatProcessor:
def __init__(self):
self.client = anthropic.Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"])
self.conversation_history = []
def process_chat_input(self, message: str, lang_code: str) -> Generator[str, None, None]:
"""
Procesa el input del chat y genera respuestas por chunks
"""
try:
# Agregar mensaje a la historia
self.conversation_history.append(f"Human: {message}")
# Generar respuesta
response = self.client.completions.create(
model="claude-3-opus-20240229",
prompt=f"{message}\n\nAssistant:",
max_tokens_to_sample=300,
temperature=0.7,
stream=True # Habilitar streaming
)
# Retornar chunks de la respuesta
full_response = ""
for chunk in response:
if chunk.completion:
yield chunk.completion
full_response += chunk.completion
# Guardar respuesta completa en la historia
self.conversation_history.append(f"Assistant: {full_response}")
except Exception as e:
logger.error(f"Error en process_chat_input: {str(e)}")
yield f"Error: {str(e)}" |