kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-14 06:07:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

unhide create_blob

Browse Source
This commit is contained in:
Patrick Devine 2025-01-13 15:22:26 -08:00
parent f718dab45d
commit a0388b2e32
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ollama/_client.py
View File

@ -559,12 +559,12 @@ class Client(BaseClient):
path = Path(args.strip()).expanduser() path = Path(args.strip()).expanduser()
path = path if path.is_absolute() else base / path path = path if path.is_absolute() else base / path
if path.exists(): if path.exists():
args = f'@{self._create_blob(path)}\n' args = f'@{self.create_blob(path)}\n'
print(command, args, end='', file=out) print(command, args, end='', file=out)
return out.getvalue() return out.getvalue()
def _create_blob(self, path: Union[str, Path]) -> str: def create_blob(self, path: Union[str, Path]) -> str:
sha256sum = sha256() sha256sum = sha256()
with open(path, 'rb') as r: with open(path, 'rb') as r:
while True: while True:
@ -1061,12 +1061,12 @@ class AsyncClient(BaseClient):
path = Path(args.strip()).expanduser() path = Path(args.strip()).expanduser()
path = path if path.is_absolute() else base / path path = path if path.is_absolute() else base / path
if path.exists(): if path.exists():
args = f'@{await self._create_blob(path)}\n' args = f'@{await self.create_blob(path)}\n'
print(command, args, end='', file=out) print(command, args, end='', file=out)
return out.getvalue() return out.getvalue()
async def _create_blob(self, path: Union[str, Path]) -> str: async def create_blob(self, path: Union[str, Path]) -> str:
sha256sum = sha256() sha256sum = sha256()
with open(path, 'rb') as r: with open(path, 'rb') as r:
while True: while True:

8
tests/test_client.py
View File

@ -622,7 +622,7 @@ def test_client_create_blob(httpserver: HTTPServer):
client = Client(httpserver.url_for('/')) client = Client(httpserver.url_for('/'))
with tempfile.NamedTemporaryFile() as blob: with tempfile.NamedTemporaryFile() as blob:
response = client._create_blob(blob.name) response = client.create_blob(blob.name)
assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855' assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'
@ -632,7 +632,7 @@ def test_client_create_blob_exists(httpserver: HTTPServer):
client = Client(httpserver.url_for('/')) client = Client(httpserver.url_for('/'))
with tempfile.NamedTemporaryFile() as blob: with tempfile.NamedTemporaryFile() as blob:
response = client._create_blob(blob.name) response = client.create_blob(blob.name)
assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855' assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'
@ -1098,7 +1098,7 @@ async def test_async_client_create_blob(httpserver: HTTPServer):
client = AsyncClient(httpserver.url_for('/')) client = AsyncClient(httpserver.url_for('/'))
with tempfile.NamedTemporaryFile() as blob: with tempfile.NamedTemporaryFile() as blob:
response = await client._create_blob(blob.name) response = await client.create_blob(blob.name)
assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855' assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'
@ -1109,7 +1109,7 @@ async def test_async_client_create_blob_exists(httpserver: HTTPServer):
client = AsyncClient(httpserver.url_for('/')) client = AsyncClient(httpserver.url_for('/'))
with tempfile.NamedTemporaryFile() as blob: with tempfile.NamedTemporaryFile() as blob:
response = await client._create_blob(blob.name) response = await client.create_blob(blob.name)
assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855' assert response == 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'