huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 20:04: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 aiohttp import ClientSession, ClientTimeout
from pydantic import ValidationError 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 ( from text_generation.types import (
StreamResponse, StreamResponse,
@ -41,7 +41,7 @@ class Client:
headers: Optional[Dict[str, str]] = None, headers: Optional[Dict[str, str]] = None,
cookies: Optional[Dict[str, str]] = None, cookies: Optional[Dict[str, str]] = None,
timeout: int = 10, timeout: int = 10,
cert: Optional[Union[str, tuple[str, str]]] = None, cert: Optional[Union[str, Tuple[str, str]]] = None,
): ):
""" """
Args: Args:
@ -290,7 +290,6 @@ class AsyncClient:
headers: Optional[Dict[str, str]] = None, headers: Optional[Dict[str, str]] = None,
cookies: Optional[Dict[str, str]] = None, cookies: Optional[Dict[str, str]] = None,
timeout: int = 10, timeout: int = 10,
cert: Optional[Union[str, tuple[str, str]]] = None,
): ):
""" """
Args: Args:
@ -302,15 +301,11 @@ class AsyncClient:
Cookies to include in the requests Cookies to include in the requests
timeout (`int`): timeout (`int`):
Timeout in seconds 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.base_url = base_url
self.headers = headers self.headers = headers
self.cookies = cookies self.cookies = cookies
self.timeout = ClientTimeout(timeout * 60) self.timeout = ClientTimeout(timeout * 60)
self.cert = cert
async def generate( async def generate(
self, self,