retoor ff9d1ae170 feat: add ws attribute and dataset support to AsyncClient chat method
- Initialize self.ws to None and assign websocket connection in __aenter__
- Close client connection and reset ws to None in __aexit__
- Extend chat method signature with optional datasets parameter, defaulting to empty list
- Pass datasets argument to underlying client.chat call
- Fix close method to set self.ws to None instead of self.client
2024-12-09 17:49:50 +00:00

Yura LLM Client for Katya server

Part of project with as target replacing the native ollama protocol. This protocol supports streaming and is usable trough https and it is possible to directly attach a web client to the backend.

Install

pip install -e .

Build

make build

Command line usage

yura ws://[host]:[port]/[path]/

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())
S
Description
Async client for the katya AI server
Readme
334 KiB
Languages
Python 96.9%
Makefile 3.1%