File size: 914 Bytes
261bb88 5f8de6f cadd679 ddb7871 5f8de6f 1d86e6d cadd679 5f8de6f 1d86e6d 5f8de6f 1d86e6d cadd679 5f8de6f 1d86e6d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from fastapi import HTTPException
from api.config import MODEL_PROVIDER_MAPPING
from api.provider import amigochat
from api.logger import setup_logger
logger = setup_logger(__name__)
async def process_streaming_response(request_data):
provider_name = MODEL_PROVIDER_MAPPING.get(request_data.get('model'))
if provider_name == 'amigochat':
return await amigochat.process_streaming_response(request_data)
else:
raise HTTPException(status_code=400, detail=f"Model {request_data.get('model')} is not supported for streaming.")
async def process_non_streaming_response(request_data):
provider_name = MODEL_PROVIDER_MAPPING.get(request_data.get('model'))
if provider_name == 'amigochat':
return await amigochat.process_non_streaming_response(request_data)
else:
raise HTTPException(status_code=400, detail=f"Model {request_data.get('model')} is not supported.")
|