| | |
|---|
| | | return |
|---|
| | | |
|---|
| | | if agent_type == AgentType.RAGFLOW: |
|---|
| | | ragflow_service = RagflowService(settings.ragflow_base_url) |
|---|
| | | ragflow_service = RagflowService(settings.fwr_base_url) |
|---|
| | | token = get_ragflow_token(db, current_user.id) |
|---|
| | | try: |
|---|
| | | async def forward_to_ragflow(): |
|---|
| | |
|---|
| | | |
|---|
| | | elif agent_type == AgentType.BISHENG: |
|---|
| | | token = get_bisheng_token(db, current_user.id) |
|---|
| | | service_uri = f"{settings.bisheng_websocket_url}/api/v1/assistant/chat/{agent_id}?t=&chat_id={chat_id}" |
|---|
| | | service_uri = f"{settings.sgb_websocket_url}/api/v1/assistant/chat/{agent_id}?t=&chat_id={chat_id}" |
|---|
| | | headers = {'cookie': f"access_token_cookie={token};"} |
|---|
| | | |
|---|
| | | async with websockets.connect(service_uri, extra_headers=headers) as service_websocket: |
|---|
