mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-01-09 15:17:59 -05:00
0b919522ae
* split execution manager and removed ns and use direct uri with k8s and docker specific dns * formating * split execution manager * refactor(builder): Fix linting warning and errors (#8021) * Fix lint errors * Fix dependency loop * address feedback * docker compose * remove ns entirely * remove yarn lock changes * update readme * remove ref * dockerfile and log * update log * debug * rename to executor * remove execution from rest * exec.py * linting * udpate tests to use config * fix test --------- Co-authored-by: Krzysztof Czerwinski <34861343+kcze@users.noreply.github.com>
37 lines
915 B
Python
37 lines
915 B
Python
import pytest
|
|
|
|
from autogpt_server.util.service import AppService, expose, get_service_client
|
|
|
|
|
|
class TestService(AppService):
|
|
def __init__(self):
|
|
super().__init__(port=8005)
|
|
self.use_redis = False
|
|
|
|
def run_service(self):
|
|
super().run_service()
|
|
|
|
@expose
|
|
def add(self, a: int, b: int) -> int:
|
|
return a + b
|
|
|
|
@expose
|
|
def subtract(self, a: int, b: int) -> int:
|
|
return a - b
|
|
|
|
@expose
|
|
def fun_with_async(self, a: int, b: int) -> int:
|
|
async def add_async(a: int, b: int) -> int:
|
|
return a + b
|
|
|
|
return self.run_and_wait(add_async(a, b))
|
|
|
|
|
|
@pytest.mark.asyncio(scope="session")
|
|
async def test_service_creation(server):
|
|
with TestService():
|
|
client = get_service_client(TestService, 8005)
|
|
assert client.add(5, 3) == 8
|
|
assert client.subtract(10, 4) == 6
|
|
assert client.fun_with_async(5, 3) == 8
|