Fix the bug where claude stream is not defined.
Browse files- request.py +3 -6
request.py
CHANGED
@@ -40,8 +40,7 @@ async def get_gemini_payload(request, engine, provider):
|
|
40 |
'Content-Type': 'application/json'
|
41 |
}
|
42 |
model = provider['model'][request.model]
|
43 |
-
|
44 |
-
gemini_stream = "streamGenerateContent"
|
45 |
url = provider['base_url']
|
46 |
if url.endswith("v1beta"):
|
47 |
url = "https://generativelanguage.googleapis.com/v1beta/models/{model}:{stream}?key={api_key}".format(model=model, stream=gemini_stream, api_key=provider['api'])
|
@@ -232,8 +231,7 @@ async def get_vertex_gemini_payload(request, engine, provider):
|
|
232 |
if provider.get("project_id"):
|
233 |
project_id = provider.get("project_id")
|
234 |
|
235 |
-
|
236 |
-
gemini_stream = "streamGenerateContent"
|
237 |
model = provider['model'][request.model]
|
238 |
location = gem
|
239 |
url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/google/models/{MODEL_ID}:{stream}".format(LOCATION=location.next(), PROJECT_ID=project_id, MODEL_ID=model, stream=gemini_stream)
|
@@ -381,8 +379,7 @@ async def get_vertex_claude_payload(request, engine, provider):
|
|
381 |
elif "claude-3-haiku" in model:
|
382 |
location = c3h
|
383 |
|
384 |
-
|
385 |
-
claude_stream = "streamRawPredict"
|
386 |
url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/anthropic/models/{MODEL}:{stream}".format(LOCATION=location.next(), PROJECT_ID=project_id, MODEL=model, stream=claude_stream)
|
387 |
|
388 |
messages = []
|
|
|
40 |
'Content-Type': 'application/json'
|
41 |
}
|
42 |
model = provider['model'][request.model]
|
43 |
+
gemini_stream = "streamGenerateContent"
|
|
|
44 |
url = provider['base_url']
|
45 |
if url.endswith("v1beta"):
|
46 |
url = "https://generativelanguage.googleapis.com/v1beta/models/{model}:{stream}?key={api_key}".format(model=model, stream=gemini_stream, api_key=provider['api'])
|
|
|
231 |
if provider.get("project_id"):
|
232 |
project_id = provider.get("project_id")
|
233 |
|
234 |
+
gemini_stream = "streamGenerateContent"
|
|
|
235 |
model = provider['model'][request.model]
|
236 |
location = gem
|
237 |
url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/google/models/{MODEL_ID}:{stream}".format(LOCATION=location.next(), PROJECT_ID=project_id, MODEL_ID=model, stream=gemini_stream)
|
|
|
379 |
elif "claude-3-haiku" in model:
|
380 |
location = c3h
|
381 |
|
382 |
+
claude_stream = "streamRawPredict"
|
|
|
383 |
url = "https://{LOCATION}-aiplatform.googleapis.com/v1/projects/{PROJECT_ID}/locations/{LOCATION}/publishers/anthropic/models/{MODEL}:{stream}".format(LOCATION=location.next(), PROJECT_ID=project_id, MODEL=model, stream=claude_stream)
|
384 |
|
385 |
messages = []
|