Spaces:
Runtime error
Runtime error
Update apis/chat_api.py
Browse files- apis/chat_api.py +2 -0
apis/chat_api.py
CHANGED
|
@@ -103,6 +103,7 @@ class ChatAPIApp:
|
|
| 103 |
):
|
| 104 |
try:
|
| 105 |
print(item.messages)
|
|
|
|
| 106 |
api_key = self.auth_api_key(api_key)
|
| 107 |
|
| 108 |
if item.model == "gpt-3.5-turbo":
|
|
@@ -188,6 +189,7 @@ class ChatAPIApp:
|
|
| 188 |
self, item: GenerateRequest, api_key: str = Depends(extract_api_key)
|
| 189 |
):
|
| 190 |
try:
|
|
|
|
| 191 |
api_key = self.auth_api_key(api_key)
|
| 192 |
|
| 193 |
if item.model == "gpt-3.5-turbo":
|
|
|
|
| 103 |
):
|
| 104 |
try:
|
| 105 |
print(item.messages)
|
| 106 |
+
item.model = "llama3-8b" if item.model == "llama3" else item.model
|
| 107 |
api_key = self.auth_api_key(api_key)
|
| 108 |
|
| 109 |
if item.model == "gpt-3.5-turbo":
|
|
|
|
| 189 |
self, item: GenerateRequest, api_key: str = Depends(extract_api_key)
|
| 190 |
):
|
| 191 |
try:
|
| 192 |
+
item.model = "llama3-8b" if item.model == "llama3" else item.model
|
| 193 |
api_key = self.auth_api_key(api_key)
|
| 194 |
|
| 195 |
if item.model == "gpt-3.5-turbo":
|