rag-gateway/app/service/v2/app_driver/chat_dialog.py

from app.service.v2.app_driver.chat_base import ChatBase


class ChatDialog(ChatBase):


    def __init__(self, token):
        self.token = token


    async def get_headers(self):
        return {
            'Content-Type': 'application/json',
            'Authorization': f'Bearer {self.token}'
        }


    async def chat_completions(self, url, data, headers):

        async for rag_response in self.http_stream(url, data, headers):

            yield rag_response