Cosmos LLM

app.py

@@ -34,7 +34,7 @@ def convert_to_messages(message, history):
 
     return messages
 
-def chat_cosmosllama(message: str, history: list, conversation_id: str, parent_message_id: str) -> (str, str):
+def chat_cosmosllm(message: str, history: list, conversation_id: str, parent_message_id: str) -> (str, str):
     messages = convert_to_messages(message, history)
 
     headers = {
@@ -63,8 +63,8 @@ def chat_cosmosllama(message: str, history: list, conversation_id: str, parent_m
                     if "assistantMessageId" in json_line:
                         new_parent_message_id = json_line["assistantMessageId"]
 
-                    if "CosmosLlama" in json_line:
-                        full_response += json_line['CosmosLlama']
+                    if "CosmosLLM" in json_line:
+                        full_response += json_line['CosmosLLM']
                         yield full_response, new_parent_message_id
                 except Exception:
                     traceback.print_exc()
@@ -77,7 +77,7 @@ def get_conversation_id():
 def clear_conversation():
     return get_conversation_id(), []
 
-chatbot = gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Cosmos LLaMa')
+chatbot = gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Cosmos LLM')
 
 with gr.Blocks(fill_height=True, css=css) as demo:
     conversation_id_state = gr.State(value=get_conversation_id())
@@ -85,7 +85,7 @@ with gr.Blocks(fill_height=True, css=css) as demo:
     
     def handle_user_message(message, history, conversation_id_state, parent_message_history):
         is_parent_message_set = False
-        response_generator = chat_cosmosllama(message, history, conversation_id_state, parent_message_history[-1] if parent_message_history else None)
+        response_generator = chat_cosmosllm(message, history, conversation_id_state, parent_message_history[-1] if parent_message_history else None)
         for response, updated_parent_message_id in response_generator:
             if not is_parent_message_set:
                 parent_message_history.append(updated_parent_message_id)