| | |
|---|
| | | ) |
|---|
| | | if response.status_code != 200: |
|---|
| | | raise Exception(f"Ragflow login failed: {response.text}") |
|---|
| | | return response.json().get('data', {}).get('access_token') |
|---|
| | | # 从响应头中提取 Authorization 字段 |
|---|
| | | authorization = response.headers.get('Authorization') |
|---|
| | | if not authorization: |
|---|
| | | raise Exception("Authorization header not found in response") |
|---|
| | | return authorization |
|---|
| | | |
|---|
| | | async def chat(self, token: str, chat_id: str, chat_history: list): |
|---|
| | | data = { |
|---|
| | | "conversation_id": chat_id, |
|---|
| | | "messages": chat_history |
|---|
| | | } |
|---|
| | | target_url = f"{self.base_url}/v1/conversation/completion" |
|---|
| | | async with httpx.AsyncClient() as client: |
|---|
| | | headers = { |
|---|
| | | 'Content-Type': 'application/json', |
|---|
| | | 'Authorization': token |
|---|
| | | } |
|---|
| | | # 创建流式请求 |
|---|
| | | async with client.stream("POST", target_url, json=data, headers=headers) as response: |
|---|
| | | # 检查响应状态码 |
|---|
| | | if response.status_code == 200: |
|---|
| | | # 流式读取响应 |
|---|
| | | async for answer in response.aiter_text(): |
|---|
| | | yield answer |
|---|
| | | else: |
|---|
| | | yield f"Error: {response.status_code}" |
|---|
| | | |
|---|
