| from huggingface_hub import HfApi, login | |
| from naptha_sdk.client.hub import Hub | |
| from naptha_sdk.client.node import Node | |
| from typing import Dict, List, Tuple | |
| class Naptha: | |
| """The entry point into Naptha.""" | |
| def __init__(self, | |
| user, | |
| hub_username, | |
| hub_password, | |
| hf_username, | |
| hf_access_token, | |
| hub_url="ws://node.naptha.ai:3001/rpc", | |
| node_url="http://node.naptha.ai:7001", | |
| routing_url=None, | |
| indirect_node_id=None, | |
| *args, | |
| **kwargs | |
| ): | |
| self.user = user | |
| self.hub_url = hub_url | |
| self.hf_username = hf_username | |
| login(hf_access_token) | |
| self.hf = HfApi() | |
| self.node_url = node_url | |
| self.routing_url = routing_url | |
| self.indirect_node_id = indirect_node_id | |
| self.node = Node( | |
| node_url=node_url, | |
| routing_url=routing_url, | |
| indirect_node_id=indirect_node_id | |
| ) | |
| self.__storedargs = user, hub_username, hub_password, hub_url, node_url, args, kwargs | |
| self.async_initialized = False | |
| async def __ainit__(self, | |
| user, | |
| hub_username, | |
| hub_password, | |
| hub_url, | |
| node_url, | |
| routing_url, | |
| indirect_node_id, | |
| *args, | |
| **kwargs): | |
| """Async constructor""" | |
| self.hub = await Hub(hub_username, hub_password, hub_url) | |
| async def __initobj(self): | |
| """Crutch used for __await__ after spawning""" | |
| assert not self.async_initialized | |
| self.async_initialized = True | |
| await self.__ainit__( | |
| self.__storedargs[0], | |
| self.__storedargs[1], | |
| self.__storedargs[2], | |
| self.__storedargs[3], | |
| self.__storedargs[4], | |
| self.__storedargs[5], | |
| self.__storedargs[6], | |
| ) | |
| return self | |
| def __await__(self): | |
| return self.__initobj().__await__() | |