This commit is contained in:
Nicolas Patry 2025-03-06 16:31:20 +01:00
parent 818c8db29a
commit 9aa71d61fb
No known key found for this signature in database
GPG Key ID: 4242CEF24CB6DBF9
3 changed files with 0 additions and 11 deletions

View File

@ -39,13 +39,6 @@ from typing import Dict, List, Optional
from aiohttp import ClientConnectorError, ClientOSError, ServerDisconnectedError
from docker.errors import NotFound
from syrupy.extensions.json import JSONSnapshotExtension
from huggingface_hub.inference._generated.types.chat_completion import (
ChatCompletionStreamOutput,
ChatCompletionOutput,
)
from openai.types.chat.chat_completion_chunk import (
ChatCompletionChunk as OAIChatCompletionChunk,
)
from text_generation import AsyncClient
from text_generation.types import (
BestOfSequence,

View File

@ -13,4 +13,3 @@ def chat_handle(launcher):
async def chat_client(chat_handle):
await chat_handle.health(300)
return chat_handle.client

View File

@ -1,10 +1,7 @@
import pytest
import requests
import json
from aiohttp import ClientSession
from openai import OpenAI
from huggingface_hub import InferenceClient
from text_generation.types import Completion
@pytest.fixture(scope="module")