Style upgrade

README.md

@@ -20,12 +20,15 @@ yura ws://[host]:[port]/[path]/
 ## Python
 ```python
 import asyncio
+
 from yura.client import AsyncClient
 
+
 async def communicate():
     client = AsyncClient("ws://[host]:[port]/[path]/")
     async for response in client.chat("Your prompt"):
         print(response)
 
+
 asyncio.run(communicate())
 ```

make

@@ -1,14 +1,16 @@
 #!/usr/bin/env python3
-import pathlib 
 import os
+import pathlib
 import sys
 
 args = sys.argv[1:]
 args_string = " ".join(args)
 
+
 def install():
     os.system("./.venv/bin/python -m pip install -e .")
 
+
 def build():
     os.system("./.venv/bin/python -m pip install build")
     os.system("rm -r dist")
@@ -17,7 +19,6 @@ def build():
     os.system("./.venv/bin/python -m black .")
 
 
-
 if not pathlib.Path(".venv").exists():
     os.system("python3 -m venv .venv")
     install()
@@ -35,4 +36,3 @@ if "publish" in sys.argv:
 
 if "run" in sys.argv:
     os.system("./.venv/bin/yura " + args_string)
-

src/yura/cli.py

@@ -1,7 +1,8 @@
-from yura.client import cli_client
 import asyncio
 import sys
 
+from yura.client import cli_client
+
 
 def run():
     try:

src/yura/client.py

@@ -1,8 +1,9 @@
 import asyncio
-import websockets
 import json
 import sys
 
+import websockets
+
 
 class AsyncRPCClient:
 
@@ -27,21 +28,23 @@ class AsyncRPCClient:
             response_raw = await ws.recv()
             response = json.loads(response_raw)
             yield response
-            if response.get('done'):
+            if response.get("done"):
                 break
 
     def __getattr__(self, name):
-        async def call(*args,**kwargs):
+        async def call(*args, **kwargs):
             ws = await self.ws
-            await ws.send(json.dumps(dict(
+            await ws.send(
-                method=name,
+                json.dumps(
-                args=args,
+                    {"method": name, "args": args, "kwargs": kwargs}, default=str
-                kwargs=kwargs
+                )
-            ),default=str))
+            )
             response = await ws.recv()
             return json.loads(response)
+
         return call
 
+
 class AsyncClient:
 
     def __init__(self, url="ws://127.0.0.1:8470"):
@@ -60,29 +63,23 @@ class AsyncClient:
         pass
 
     async def create(self, name, extends, system):
-        return await self.client.create(
+        return await self.client.create(name=name, extends=extends, system=system)
-            name=name,
-            extends=extends,
-            system=system
-        )
 
-    async def chat(self, token,message):
+    async def chat(self, token, message):
-        yield await self.client.chat(uid=token,message=message)
+        yield await self.client.chat(uid=token, message=message)
         async for msg in self.client:
             yield msg
-            if msg.get('done'):
+            if msg.get("done"):
                 break
 
-    async def connect(self,name):
+    async def connect(self, name):
         return await self.client.connect(name)
 
 
-
 async def cli_client(url="ws://127.0.0.1:8470"):
 
     loop = asyncio.get_event_loop()
-    async_client = AsyncClient(url)
+    AsyncClient(url)
-
 
     async with AsyncClient(url) as client:
         name = "retoor3b"
@@ -99,14 +96,14 @@ async def cli_client(url="ws://127.0.0.1:8470"):
         #     """
         # )
         token = await client.connect(name)
-        #print(success)
+        # print(success)
         print(token)
         while True:
 
             sys.stdout.write("> ")
             sys.stdout.flush()
             message_content = await loop.run_in_executor(None, sys.stdin.readline)
-            #message_content="AAA"
+            # message_content="AAA"
             async for response in client.chat(token, message_content):
 
                 print(response["content"], end="", flush=True)