ziya 0.1.6__tar.gz → 0.1.8__tar.gz

{ziya-0.1.6 → ziya-0.1.8}/PKG-INFO

@@ -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 → ziya-0.1.8}/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

{ziya-0.1.6 → ziya-0.1.8}/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"

{ziya-0.1.6 → ziya-0.1.8}/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>
         ))}