commandchat 0.0.10__tar.gz → 0.0.12__tar.gz

{commandchat-0.0.10 → commandchat-0.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commandchat
-Version: 0.0.10
+Version: 0.0.12
 Summary: use command to chat with openai models
 Home-page: https://github.com/
 Author: xoto

{commandchat-0.0.10 → commandchat-0.0.12}/commandchat.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commandchat
-Version: 0.0.10
+Version: 0.0.12
 Summary: use command to chat with openai models
 Home-page: https://github.com/
 Author: xoto

{commandchat-0.0.10 → commandchat-0.0.12}/occ/CommandChat.py

@@ -1,26 +1,23 @@
 import asyncio
 import json
+import os
 import sys
 import time
 from pathlib import Path
 from typing import AsyncGenerator
 
-import openai
-from openai import OpenAI, Stream
 from openai import AzureOpenAI
-import os
-
-from openai.types.chat import ChatCompletionChunk
+from openai import OpenAI
 from openai.types.chat.chat_completion_chunk import Choice
 from prompt_toolkit import print_formatted_text, HTML, Application
+from prompt_toolkit.clipboard.pyperclip import PyperclipClipboard
 from prompt_toolkit.layout import Layout, HSplit
 from prompt_toolkit.widgets import TextArea
 from rich.console import Console
-from rich.markdown import Markdown
 from rich.live import Live
+from rich.markdown import Markdown
 
 from occ.commons.config import get_env
-from occ.utils.CommonUtil import save_and_copy_image, waiting_start, waiting_stop
 
 DEFAULT_CHAT_LOG_ID = "chat-1"
 DEFAULT_PROFILE = "default"
@@ -37,9 +34,9 @@ def get_home_path():
     return homedir
 
 
+clip = PyperclipClipboard()
 console = Console()
 
-
 def print_formatted(content: str, live: Live):
     md = Markdown(content)
     live.update(md)
@@ -99,6 +96,7 @@ class CommandChat:
                 for choice in completion.choices:
                     completion_text += choice.text
                     print_formatted(completion_text, live)
+        clip.set_text(completion_text)
         print("\n")
 
     def chat_completions(self, message, model):
@@ -108,7 +106,7 @@ class CommandChat:
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
         try:
-            final_text = loop.run_until_complete(self.main_with_stream(self.async_stream))
+            final_text = loop.run_until_complete(self.print_streaming(self.async_stream))
         except KeyboardInterrupt:
             final_text = None
         finally:
@@ -119,8 +117,8 @@ class CommandChat:
             sys.exit(0)
         md = Markdown(final_text)
         self.append_to_history(final_text)
-        console.clear()
         console.print(md)
+        clip.set_text(final_text)
         self.record_chat_logs(message, {"role": self.role, "content": final_text.replace("\n\n", "")})
 
     async def async_stream(self) -> AsyncGenerator[Choice, None]:
@@ -139,7 +137,7 @@ class CommandChat:
             await asyncio.sleep(0.01)
             yield choice
 
-    async def main_with_stream(self, async_stream):
+    async def print_streaming(self, async_stream):
         self.partial_text = []
         text_area = TextArea(
             text="",
@@ -208,4 +206,4 @@ class CommandChat:
 
 if __name__ == '__main__':
     command_chat = CommandChat()
-    command_chat.chat("帮我写一个python的冒泡排序算法, 详细点回答", "o1-mini")
+    command_chat.chat("帮我写一个python的冒泡排序算法", "o1-mini")

{commandchat-0.0.10 → commandchat-0.0.12}/occ/command/__main__.py

@@ -53,7 +53,6 @@ def chat(message, id, profile, model, file):
                 style=style_from_pygments_cls(TangoStyle), multiline=True, wrap_lines=True,
                 cursor=ModalCursorShapeConfig(),
             )
-            print_formatted_text(HTML("<ansibrightwhite>AI model Terminal</ansibrightwhite>\n"))
             while True:
                 try:
                     message = session.prompt("👤 You: \n")

{commandchat-0.0.10 → commandchat-0.0.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "commandchat"
-version = "0.0.10"
+version = "0.0.12"
 description = "use command to chat with openai models"
 readme = "README.md"
 requires-python = ">=3.8"