Update chain_app.py

chain_app.py

@@ -1030,6 +1030,34 @@ async def chat_profile():
                 
             ]
         ),
+        cl.ChatProfile(
+            name='Yi-1.5-34B',
+            markdown_description = 'Yi-1.5 34B from 01-AI',
+            starters=[
+                cl.Starter(
+                    label="Morning routine ideation",
+                    message="Can you help me create a personalized morning routine that would help increase my productivity throughout the day? Start by asking me about my current habits and what activities energize me in the morning.",
+                    icon="https://chainlit-rag-copilot-r2xd.onrender.com/public/idea.svg",
+                    ),
+        
+                cl.Starter(
+                    label="Explain superconductors",
+                    message="Explain superconductors like I'm five years old.",
+                    icon="https://chainlit-rag-copilot-r2xd.onrender.com/public/learn.svg",
+                    ),
+                cl.Starter(
+                    label="Python script for daily email reports",
+                    message="Write a script to automate sending daily email reports in Python, and walk me through how I would set it up.",
+                    icon="https://chainlit-rag-copilot-r2xd.onrender.com/public/terminal.svg",
+                    ),
+                cl.Starter(
+                    label="Text inviting friend to wedding",
+                    message="Write a text asking a friend to be my plus-one at a wedding next month. I want to keep it super short and casual, and offer an out.",
+                    icon="https://chainlit-rag-copilot-r2xd.onrender.com/public/write.svg",
+                    )
+                
+            ]
+        ),
         # # cl.ChatProfile(
         # #     name="Toka-353M",
         # #     markdown_description="PartAI Open Source model Toka with 353M parameters",
@@ -1975,6 +2003,26 @@ async def on_chat_start():
                 ),
             ]
         ).send()
+
+    if chat_profile == 'Yi-1.5-34B':
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="01AI-Model",
+                    label="01AI - Model",
+                    values=["Yi-1.5-34B"],
+                    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 Mistral nemo 12B .i was configured by Artin Daneshvar and Sadra Noadoust, 2 iranian students to help you, how can i assist you today ? "
         # ).send()
@@ -2530,7 +2578,7 @@ async def main(message: cl.Message):
             token=f"{hf_token_llama_3_1}",
         )
         result = client.chat_completion(
-        	messages=[{"role": "user", "content": "What is the capital of France?"}],
+        	messages=[{"role": "user", "content": message.content}],
             max_tokens=500,
         )
 
@@ -2538,7 +2586,18 @@ async def main(message: cl.Message):
             content=result.choices[0].message.content,
         ).send()
 
-
+    elif chat_profile == 'Yi-1.5-34B':
+        client = InferenceClient(
+            "01-ai/Yi-1.5-34B-Chat",
+            token=f'{hf_token_llama_3_1}',
+        )
+        result = client.chat_completion(
+        	messages=[{"role": "user", "content": message.content}],
+            max_tokens=500,
+        )
+        await cl.Message(
+            content=result.choices[0].message.content,
+        ).send()
 
     # elif chat_profile == 'Toka-353M':
     #     output = query({