Update app.py

app.py

@@ -13,7 +13,7 @@ openai_models = ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-
                      "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613", "text-davinci-003", 
                      "text-davinci-002", "text-curie-001", "text-babbage-001", "text-ada-001"]
 
-google_palm_models = ["models/text-bison-001"]
+google_palm_models = ["models/text-bison-001", "models/chat-bison-001","models/embedding-gecko-001"]
 
 azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
 azure_deployment_name = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")
@@ -26,6 +26,7 @@ def openai_text_completion(prompt: str, model: str):
  
         assistant_prompt: str = None,
 
+
         messages = [
             {"role": "user", "content": f"{prompt}"},
             {"role": "system", "content": f"{system_prompt}"},