Update chain_app.py

chain_app.py

@@ -140,6 +140,10 @@ async def chat_profile():
             name="Llama-3-70B",
             markdown_description="Meta Open Source model Llama-3 with 70B parameters",
         ),
+        cl.ChatProfile(
+            name='Aya-35B',
+            markdown_description='Cohere open sourced AI model with 35B parameters'
+        ),
         cl.ChatProfile(
             name='Aya-23B',
             markdown_description='Cohere open sourced AI model with 23B parameters'
@@ -148,7 +152,32 @@ async def chat_profile():
             nanm='Command-R-Plus',
             markdown_description='Cohere open sourced AI model named Command R +'
 
-        )
+        ),
+        cl.ChatProfile(
+            nanm='Command-R',
+            markdown_description='Cohere open sourced AI model named Command R'
+
+        ),
+        cl.ChatProfile(
+            nanm='Command-Light',
+            markdown_description='Cohere open sourced AI model named Command R'
+
+        ),
+        cl.ChatProfile(
+            nanm='Command-Light-Nightly',
+            markdown_description='Cohere open sourced AI model named Command R'
+
+        ),
+        cl.ChatProfile(
+            nanm='Command-Nightly',
+            markdown_description='Cohere open sourced AI model named Command R'
+
+        ),
+        cl.ChatProfile(
+            nanm='Command',
+            markdown_description='Cohere open sourced AI model named Command R'
+
+        ),
         cl.ChatProfile(
             name="Llama-3-8B",
             markdown_description="Meta Open Source model Llama-2 with 7B parameters",
@@ -769,8 +798,31 @@ async def on_chat_start():
         ).send() 
         await cl.Message(
             content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
-        )
+        ).send()
 
+    if chat_profile == 'Aya-35B':
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Cohere-Model",
+                    label="Cohere - Model",
+                    values=["Aya-35B"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send() 
+        await cl.Message(
+            content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
+        ).send()
+        
     if chat_profile == 'Command-R-Plus:
         await cl.ChatSettings(
             [
@@ -790,9 +842,124 @@ async def on_chat_start():
                 ),
             ]
         ).send() 
+        await cl.Message(
+            content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
+        ).send()
+
+    if chat_profile == 'Command-Nigtly:
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Cohere-Model",
+                    label="Cohere - Model",
+                    values=["Command-Nigtly"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send() 
+        await cl.Message(
+            content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
+        ).send()
+
+    if chat_profile == 'Command-Light-Nigtly:
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Cohere-Model",
+                    label="Cohere - Model",
+                    values=["Command-Light-Nigtly"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send() 
+        await cl.Message(
+            content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
+        ).send()
+
+    if chat_profile == 'Command-Light:
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Cohere-Model",
+                    label="Cohere - Model",
+                    values=["Command-Light"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send() 
         await cl.Message(
             content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
         )
+
+    if chat_profile == 'Command-R:
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Cohere-Model",
+                    label="Cohere - Model",
+                    values=["Command-R"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send() 
+        await cl.Message(
+            content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
+        ).send()
+
+    if chat_profile == 'Command:
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="Cohere-Model",
+                    label="Cohere - Model",
+                    values=["Command"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send() 
+        await cl.Message(
+            content='Im one of the best open source models that cohere released. i am configured by 2 iranian boys named Artin Daneshvar and Sadra Noadosut to help you out!'
+        ).send()
         
     if chat_profile == 'gemma2-9B':
         await cl.ChatSettings(
@@ -1414,6 +1581,22 @@ async def main(message: cl.Message):
                 complete_content += event.text
         await cl.Message(content=complete_content).send()
 
+    elif chat_profile == 'Aya-35B':
+        stream = co.chat_stream( 
+        model='c4ai-aya-23',
+        message=message.content,
+        temperature=0.3,
+        # chat_history=[{"role": "User", "message": "Hello"}, {"role": "Chatbot", "message": "Hello! How can I help you today?"}, {"role": "User", "message": "Hi"}, {"role": "User", "message": "hello"}],
+        prompt_truncation='OFF',
+        connectors=[],
+        ) 
+
+        complete_content = ''
+        for event in stream:
+            if event.event_type == 'text-generation':
+                complete_content += event.text
+        await cl.Message(content=complete_content).send()
+
     elif chat_profile == 'Command-R-Plus':
         stream = co.chat_stream( 
           model='command-r-plus',
@@ -1428,6 +1611,75 @@ async def main(message: cl.Message):
                 complete_content += event.text
         await cl.Message(content=complete_content).send()
 
+    elif chat_profile == 'Command-R':
+        stream = co.chat_stream( 
+          model='command-r',
+          message=message.content,
+          temperature=0.3,
+          chat_history=[],
+          prompt_truncation='AUTO',
+        ) 
+        complete_content = ''
+        for event in stream:
+            if event.event_type == 'text-generation':
+                complete_content += event.text
+        await cl.Message(content=complete_content).send()
+
+    elif chat_profile == 'Command':
+        stream = co.chat_stream( 
+          model='command',
+          message=message.content,
+          temperature=0.3,
+          chat_history=[],
+          prompt_truncation='AUTO',
+        ) 
+        complete_content = ''
+        for event in stream:
+            if event.event_type == 'text-generation':
+                complete_content += event.text
+        await cl.Message(content=complete_content).send()
+        
+    elif chat_profile == 'Command-Light':
+        stream = co.chat_stream( 
+          model='command-light',
+          message=message.content,
+          temperature=0.3,
+          chat_history=[],
+          prompt_truncation='AUTO',
+        ) 
+        complete_content = ''
+        for event in stream:
+            if event.event_type == 'text-generation':
+                complete_content += event.text
+        await cl.Message(content=complete_content).send()
+
+    elif chat_profile == 'Command-Light-Nightly':
+        stream = co.chat_stream( 
+          model='command-light-nightly',
+          message=message.content,
+          temperature=0.3,
+          chat_history=[],
+          prompt_truncation='AUTO',
+        ) 
+        complete_content = ''
+        for event in stream:
+            if event.event_type == 'text-generation':
+                complete_content += event.text
+        await cl.Message(content=complete_content).send()
+
+    elif chat_profile == 'Command-Nightly':
+        stream = co.chat_stream( 
+          model='command-light-nightly',
+          message=message.content,
+          temperature=0.3,
+          chat_history=[],
+          prompt_truncation='AUTO',
+        ) 
+        complete_content = ''
+        for event in stream:
+            if event.event_type == 'text-generation':
+                complete_content += event.text
+        await cl.Message(content=complete_content).send()
 
 @cl.on_settings_update
 async def setup_agent(settings):