jonathanjordan21 commited on
Commit
765a3f6
1 Parent(s): 9dffcaf

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. 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":