File size: 2,024 Bytes
4ae0b03 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
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__()
|