from . import client

import asyncio

import os


class TestDifyClient:
    async def test_chat_messages(self):
        cln = client.DifyClient(api_key=os.getenv('DIFY_API_KEY'))

        resp = await cln.chat_messages(inputs={}, query='Who are you?', user_id='test')
        print(resp)


if __name__ == '__main__':
    asyncio.run(TestDifyClient().test_chat_messages())