ziya 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

app/agents/agent.py

@@ -38,7 +38,7 @@ logger.info(f"Using Claude Model: {model_id}")
 
 model = ChatBedrock(
     model_id=model_id,
-    model_kwargs={"max_tokens": 4096},
+    model_kwargs={"max_tokens": 4096, "temperature": 0.3, "top_k": 15},
     credentials_profile_name=aws_profile if aws_profile else None,
     config=botocore.config.Config(
         read_timeout=900

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ziya"
-version = "0.1.6"
+version = "0.1.8"
 description = ""
 authors = ["Vishnu Krishnaprasad <vishnukool@gmail.com>"]
 readme = "README.md"

static/app.js

@@ -13,12 +13,12 @@ const App = () => {
         }
     }, []);
 
-    const handleSendPayload = async () => {
+    const handleSendPayload = async (msgs, msg) => {
         setQuestion('');
         setIsStreaming(true);
-        setMessages((messages) => [...messages, {content: question, role: 'human'}]);
+        setMessages((messages) => [...msgs, {content: msg, role: 'human'}]);
         setStreamedContent('');
-        await window.sendPayload(messages, question, setStreamedContent, setIsStreaming);
+        await window.sendPayload(messages, msg, setStreamedContent, setIsStreaming);
         setStreamedContent((cont) => {
             setMessages((messages) => [...messages, {content: cont, role: 'assistant'}]);
             return ""
@@ -37,24 +37,78 @@ const App = () => {
                     inputTextareaRef={inputTextareaRef}
                     question={question}
                     handleChange={handleChange}
-                    sendPayload={handleSendPayload}
+                    sendPayload={() => handleSendPayload(messages, question)}
                     isStreaming={isStreaming}
                 />
                 {streamedContent || messages.length > 0 ? <div className="chat-container">
                     {streamedContent && <StreamedContent streamedContent={streamedContent}/>}
-                    {messages.length > 0 && <ChatContainer messages={messages}/>}
+                    {messages.length > 0 &&
+                        <ChatContainer messages={messages} setMessages={setMessages}
+                                       handleSendPayload={handleSendPayload}/>}
                 </div> : null}
+
             </div>
         </div>
     );
 };
 
-const ChatContainer = ({messages}) => (
+const EditButton = ({message, index, setMessages, handleSendPayload}) => {
+    const [isEditing, setIsEditing] = useState(false);
+    const [editedMessage, setEditedMessage] = useState(message.content);
+
+    const handleEdit = () => {
+        setIsEditing(true);
+    };
+
+    const handleCancel = () => {
+        setIsEditing(false);
+        setEditedMessage(message.content);
+    };
+
+    const handleSubmit = () => {
+        setIsEditing(false);
+        setMessages((prevMessages) => {
+            const updatedMessages = [...prevMessages];
+            updatedMessages.splice(index);
+            handleSendPayload(updatedMessages, editedMessage);
+            return updatedMessages;
+        });
+    };
+
+    return (
+        <div>
+            {isEditing ? (
+                <>
+                    <textarea
+                        value={editedMessage}
+                        onChange={(e) => setEditedMessage(e.target.value)}
+                    />
+                    <button onClick={handleSubmit}>Submit</button>
+                    <button onClick={handleCancel}>Cancel</button>
+                </>
+            ) : (
+                <button className="edit-button" onClick={handleEdit}>Edit</button>
+            )}
+        </div>
+    );
+};
+const ChatContainer = ({messages, setMessages, handleSendPayload}) => (
     <div>
         {messages.slice().reverse().map((msg, index) => (
             <div key={index} className={`message ${msg.role}`}>
-                {msg.role === 'human' ? <div className="message-sender">You:</div> :
-                    <div className="message-sender">AI:</div>}
+                {msg.role === 'human' ? (
+                    <div style={{display: 'flex', justifyContent: 'space-between'}}>
+                        <div className="message-sender">You:</div>
+                        <EditButton
+                            message={msg}
+                            index={messages.length - 1 - index}
+                            setMessages={setMessages}
+                            handleSendPayload={handleSendPayload}
+                        />
+                    </div>
+                ) : (
+                    <div className="message-sender">AI:</div>
+                )}
                 <MarkdownRenderer markdown={msg.content}/>
             </div>
         ))}

{ziya-0.1.6.dist-info → ziya-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.6
+Version: 0.1.8
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com

{ziya-0.1.6.dist-info → ziya-0.1.8.dist-info}/RECORD

@@ -1,6 +1,6 @@
 app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 app/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-app/agents/agent.py,sha256=Qb1xgZ7fkpLSbKzSsygze7UL2H8jMtnOiTD-Gm8uXWU,4349
+app/agents/agent.py,sha256=R_dlvm9r40sVh9jK6nJOrq-FElU5fsbx7EkhVR94rXM,4382
 app/agents/prompts.py,sha256=O54rH2KupyY7ZSjPEvm3j9DSGjdL3jJOz8F9PTcePQQ,1674
 app/main.py,sha256=DHY_hRbnGAsILHac8uJWdJJAg-9HGr6Dl6A27RnBfK4,3146
 app/server.py,sha256=YTEHYUsFQhgkGRHZ_y7K3Ju3SmUIbVxLSdgrgi1BuEU,676
@@ -11,14 +11,14 @@ app/utils/langchain_validation_util.py,sha256=RgmKayKMApvUA7SPF_DrBAcYdIzzLUVJfn
 app/utils/logging_utils.py,sha256=8JEcc1t7L-a0G4HLmM8zFydNNSOd5l2-gkTxviLQUns,280
 app/utils/print_tree_util.py,sha256=O4Rb-GS8UODxfH7Z6bZGsr-opG6QLmvdaU1im5kh4IA,1419
 app/utils/version_util.py,sha256=NcvMWIImJDcO9K5OMX6jJD4q6Zb8Y117Mu8sWIkhxEc,631
-pyproject.toml,sha256=MZq9r6D3Jr1EPgOD4C6kriMoCpgrUeZO2ZYXX_VoBOk,716
-static/app.js,sha256=qz2F0e5LxuVFI3MXWvcON9l1cN0J_ODdQOwbgT3ZqWA,3622
+pyproject.toml,sha256=jXsQJf7Tt_VYxssmnTZyv8Bmy_u1Cn4SAjfwfA2wxLQ,716
+static/app.js,sha256=E5O8rtrB1N2bjjKR9SwBbxy5ro8MawzEMU4JUeT-9W4,5513
 static/favicon.ico,sha256=HgB8xAZdDHFK2lODUsp2H_Dds_i14pnpydx7NEJrNrU,15086
 static/sendPayload.js,sha256=VnypSWF7t8bv83026LGU3GMyr9tMrp05hmWATcbCG4k,1956
 static/ziya.css,sha256=OaieSXw_7LOk0WNNi-1mPsAaS3rOxhHycKFgFDoiZB0,1395
 templates/index.html,sha256=8auy9lGLdcDzBkzpjYB1ygrAVbXpo0rqQ8p1FgMEeck,800
-ziya-0.1.6.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
-ziya-0.1.6.dist-info/METADATA,sha256=DmNfRTAI_cgCxdsh5bM3a3Vrj8nhlXm_ta488VCDfyA,2921
-ziya-0.1.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-ziya-0.1.6.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
-ziya-0.1.6.dist-info/RECORD,,
+ziya-0.1.8.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
+ziya-0.1.8.dist-info/METADATA,sha256=kkIQo33M-WwZChmNQFxkiBStVNMOA5YwWEUiPTPjbP4,2921
+ziya-0.1.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+ziya-0.1.8.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
+ziya-0.1.8.dist-info/RECORD,,