Fix Claude response reception logic

.gitignore

@@ -1,4 +1,5 @@
 api.json
 api.yaml
 .env
-__pycache__
+__pycache__
+.vscode

main.py

@@ -64,13 +64,15 @@ async def process_request(request: RequestModel, provider: Dict):
     print("provider: ", provider['provider'])
     url = provider['base_url']
     parsed_url = urlparse(url)
+    # print(parsed_url)
     engine = None
     if parsed_url.netloc == 'generativelanguage.googleapis.com':
         engine = "gemini"
-    elif parsed_url.netloc == 'api.anthropic.com':
+    elif parsed_url.netloc == 'api.anthropic.com' or parsed_url.path.endswith("v1/message"):
         engine = "claude"
     else:
         engine = "gpt"
+    print(engine)
 
     url, headers, payload = await get_payload(request, engine, provider)
 

response.py

@@ -83,12 +83,16 @@ async def fetch_claude_response_stream(client, url, headers, payload, model):
     try:
         timestamp = datetime.timestamp(datetime.now())
         async with client.stream('POST', url, headers=headers, json=payload) as response:
+            buffer = ""
             async for chunk in response.aiter_bytes():
-                chunk_line = chunk.decode('utf-8').split("\n")
-                for chunk in chunk_line:
-                    if chunk.startswith("data:"):
-                        line = chunk[6:]
+                buffer += chunk.decode('utf-8')
+                while "\n" in buffer:
+                    line, buffer = buffer.split("\n", 1)
+                    print(line)
+
+                    if line.startswith("data:"):
                         print(line)
+                        line = line[6:]
                         resp: dict = json.loads(line)
                         message = resp.get("message")
                         if message:

test/launch_with_clear.py

@@ -0,0 +1,8 @@
+import os
+import subprocess
+
+# 清理终端
+subprocess.call('clear' if os.name == 'posix' else 'cls', shell=True)
+
+# 运行主程序
+os.system(os.environ['PYTHONPATH'] + ' main.py')