mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-05-08 03:01:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

test(memory): harden OpenViking local upload coverage

Browse source
This commit is contained in:
Hao Zhe 2026-05-04 15:38:28 +08:00 • committed by Teknium
parent 7137cccbd1
commit 187951ec6b
2 changed files with 96 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
plugins/memory/openviking/__init__.py
View file

@ -110,14 +110,10 @@ class _VikingClient:
def _url(self, path: str) -> str:
return f"{self._endpoint}{path}"
def _auth_headers(self) -> dict:
h = {
"X-OpenViking-Account": self._account,
"X-OpenViking-User": self._user,
}
if self._api_key:
h["X-API-Key"] = self._api_key
return h
def _multipart_headers(self) -> dict:
headers = self._headers()
headers.pop("Content-Type", None)
return headers
def _parse_response(self, resp) -> dict:
try:
@ -167,7 +163,7 @@ class _VikingClient:
resp = self._httpx.post(
self._url("/api/v1/resources/temp_upload"),
files={"file": (file_path.name, f, mime_type)},
headers=self._auth_headers(),
headers=self._multipart_headers(),
timeout=_TIMEOUT,
)
data = self._parse_response(resp)