Spaces:
Running
Running
Add optional param headers
Browse files
horde.py
CHANGED
@@ -3,16 +3,15 @@ import httpx
|
|
3 |
|
4 |
hordeApiKey = os.environ.get("HORDE_API_KEY")
|
5 |
baseUrl = "https://aihorde.net/api/v2/"
|
|
|
|
|
|
|
|
|
6 |
|
7 |
class HordeAPI:
|
8 |
# 省去类初始化,直接在每个方法里创建客户端
|
9 |
|
10 |
-
async def _fetch(self, url, method="GET"):
|
11 |
-
headers = {
|
12 |
-
"Content-Type": "application/json",
|
13 |
-
"apikey": f"{hordeApiKey}"
|
14 |
-
}
|
15 |
-
|
16 |
async with httpx.AsyncClient() as client:
|
17 |
if method == "GET":
|
18 |
response = await client.get(url, headers=headers)
|
@@ -51,12 +50,12 @@ class HordeAPI:
|
|
51 |
def generateCheck(id):
|
52 |
async def api_call():
|
53 |
url = f"{baseUrl}generate/check/{id}"
|
54 |
-
return await HordeAPI()._fetch(url)
|
55 |
return HordeAPI.APIContextManager(api_call)
|
56 |
|
57 |
@staticmethod
|
58 |
def generateStatus(id):
|
59 |
async def api_call():
|
60 |
url = f"{baseUrl}generate/status/{id}"
|
61 |
-
return await HordeAPI()._fetch(url)
|
62 |
return HordeAPI.APIContextManager(api_call)
|
|
|
3 |
|
4 |
hordeApiKey = os.environ.get("HORDE_API_KEY")
|
5 |
baseUrl = "https://aihorde.net/api/v2/"
|
6 |
+
headers = {
|
7 |
+
"Content-Type": "application/json",
|
8 |
+
"apikey": f"{hordeApiKey}"
|
9 |
+
}
|
10 |
|
11 |
class HordeAPI:
|
12 |
# 省去类初始化,直接在每个方法里创建客户端
|
13 |
|
14 |
+
async def _fetch(self, url, method="GET", headers=headers):
|
|
|
|
|
|
|
|
|
|
|
15 |
async with httpx.AsyncClient() as client:
|
16 |
if method == "GET":
|
17 |
response = await client.get(url, headers=headers)
|
|
|
50 |
def generateCheck(id):
|
51 |
async def api_call():
|
52 |
url = f"{baseUrl}generate/check/{id}"
|
53 |
+
return await HordeAPI()._fetch(url, headers={"Content-Type": "application/json"})
|
54 |
return HordeAPI.APIContextManager(api_call)
|
55 |
|
56 |
@staticmethod
|
57 |
def generateStatus(id):
|
58 |
async def api_call():
|
59 |
url = f"{baseUrl}generate/status/{id}"
|
60 |
+
return await HordeAPI()._fetch(url, headers={"Content-Type": "application/json"})
|
61 |
return HordeAPI.APIContextManager(api_call)
|