huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 11:54:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1 from bbc/fix-type-hint

Browse Source 
Fix type hint and remove unused cert
This commit is contained in:
Matt Haynes 2023-07-14 12:06:04 +01:00 committed by GitHub
commit 152085461d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clients/python/text_generation/client.py
View File

@ -3,7 +3,7 @@ import requests
from aiohttp import ClientSession, ClientTimeout
from pydantic import ValidationError
from typing import Dict, Optional, List, AsyncIterator, Iterator, Union
from typing import Dict, Optional, List, AsyncIterator, Iterator, Union, Tuple
from text_generation.types import (
StreamResponse,
@ -41,7 +41,7 @@ class Client:
headers: Optional[Dict[str, str]] = None,
cookies: Optional[Dict[str, str]] = None,
timeout: int = 10,
cert: Optional[Union[str, tuple[str, str]]] = None,
cert: Optional[Union[str, Tuple[str, str]]] = None,
):
"""
Args:
@ -290,7 +290,6 @@ class AsyncClient:
headers: Optional[Dict[str, str]] = None,
cookies: Optional[Dict[str, str]] = None,
timeout: int = 10,
cert: Optional[Union[str, tuple[str, str]]] = None,
):
"""
Args:
@ -302,15 +301,11 @@ class AsyncClient:
Cookies to include in the requests
timeout (`int`):
Timeout in seconds
cert (`Optional[Union[str, tuple[str, str]]]`):
If String, path to ssl client cert file (.pem).
If Tuple, ('cert', 'key') pair.
"""
self.base_url = base_url
self.headers = headers
self.cookies = cookies
self.timeout = ClientTimeout(timeout * 60)
self.cert = cert
async def generate(
self,