Spaces:
Running
Running
jonathanjordan21
commited on
Commit
•
765a3f6
1
Parent(s):
9dffcaf
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":
|