Initial commit

src/yura.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.1
+Name: yura
+Version: 13.3.7
+Summary: Yura async AI client
+Author: retoor
+Author-email: retoor@retoor.io
+License: MIT
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: websockets

src/yura.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+pyproject.toml
+setup.cfg
+src/yura.egg-info/PKG-INFO
+src/yura.egg-info/SOURCES.txt
+src/yura.egg-info/dependency_links.txt
+src/yura.egg-info/entry_points.txt
+src/yura.egg-info/requires.txt
+src/yura.egg-info/top_level.txt

src/yura.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

src/yura.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+yura = yura.client:main

src/yura.egg-info/requires.txt

@@ -0,0 +1 @@
+websockets

src/yura.egg-info/top_level.txt

@@ -0,0 +1 @@
+

src/yura/client.py

@@ -0,0 +1,106 @@
+import asyncio
+import websockets
+import json
+import sys
+
+
+class AsyncClient:
+
+    def __init__(self, url="ws://127.0.0.1:8470"):
+
+        self.url = url
+        self.ws = None
+        self.queue_in = asyncio.Queue()
+        self.queue_out = asyncio.Queue()
+        self.communication_task = None
+
+    async def ensure_connection():
+
+        if not self.ws:
+            self.ws = await websockets.connect(self.url)
+
+        return self.ws
+
+    async def ensure_communication(self):
+
+        if not self.communication_task:
+            self.communication_task = asyncio.create_task(self.communicate())
+
+        return self.communication_task
+
+    async def chat(self, message):
+        await self.ensure_communication()
+        await self.queue_out.put(message)
+        while True:
+
+            while True:
+                try:
+                    response = await asyncio.wait_for(self.queue_in.get(), 0.1)
+
+                except asyncio.TimeoutError:
+                    continue
+                    break
+
+            yield response
+
+            if response["done"]:
+                break
+
+    async def communicate(self):
+        loop = asyncio.get_event_loop()
+        async with websockets.connect(self.url) as websocket:
+            while True:
+                message_content = None
+                while not message_content:
+                    try:
+                        message_content = await asyncio.wait_for(
+                            self.queue_out.get(), 0.1
+                        )
+                    except asyncio.TimeoutError:
+                        continue
+
+                response = await websocket.send(json.dumps(message_content))
+
+                while True:
+                    response = json.loads(await websocket.recv())
+
+                    if response["done"]:
+                        break
+
+                    await self.queue_in.put(response)
+
+                await self.queue_in.put(response)
+
+
+async def cli_client(url="ws://127.0.0.1:8470"):
+
+    loop = asyncio.get_event_loop()
+    async_client = AsyncClient(url)
+
+    while True:
+
+        sys.stdout.write("> ")
+        sys.stdout.flush()
+        message_content = await loop.run_in_executor(None, sys.stdin.readline)
+
+        async for response in async_client.chat(message_content):
+
+            print(response["content"], end="", flush=True)
+
+            if response["done"]:
+                break
+
+        print("")
+
+
+def main():
+    url = "ws://127.0.0.1:8470"
+    try:
+        url = sys.argv[1]
+    except IndexError:
+        pass
+    asyncio.run(cli_client(url))
+
+
+if __name__ == "__main__":
+    main()