From 681f2fc1d3a3c3ca48afe3abcde4cb62ace78eb4 Mon Sep 17 00:00:00 2001 From: luc Date: Sat, 23 Mar 2024 10:44:59 +0100 Subject: [PATCH] pre-commit formatting --- genai_impact/tracers/mistralai_tracer.py | 4 ++-- genai_impact/tracers/openai_tracer.py | 5 ++--- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/genai_impact/tracers/mistralai_tracer.py b/genai_impact/tracers/mistralai_tracer.py index ce5c041..da3eb61 100644 --- a/genai_impact/tracers/mistralai_tracer.py +++ b/genai_impact/tracers/mistralai_tracer.py @@ -6,8 +6,8 @@ from genai_impact.model_repository import models try: - from mistralai.client import MistralClient as _MistralClient from mistralai.async_client import MistralAsyncClient as _MistralAsyncClient + from mistralai.client import MistralClient as _MistralClient from mistralai.models.chat_completion import ( ChatCompletionResponse as _ChatCompletionResponse, ) @@ -46,7 +46,7 @@ async def mistralai_async_chat_wrapper( wrapped: Callable, instance: _MistralAsyncClient, args: Any, - kwargs: Any, # noqa: ARG001 + kwargs: Any, ) -> ChatCompletionResponse: response = await wrapped(*args, **kwargs) return compute_impacts_and_return_response(response) diff --git a/genai_impact/tracers/openai_tracer.py b/genai_impact/tracers/openai_tracer.py index c62b61b..ed8bba0 100644 --- a/genai_impact/tracers/openai_tracer.py +++ b/genai_impact/tracers/openai_tracer.py @@ -1,7 +1,6 @@ from typing import Any, Callable -from openai.resources.chat import Completions, AsyncCompletions - +from openai.resources.chat import AsyncCompletions, Completions from openai.types.chat import ChatCompletion as _ChatCompletion from wrapt import wrap_function_wrapper @@ -36,7 +35,7 @@ async def openai_async_chat_wrapper( wrapped: Callable, instance: AsyncCompletions, args: Any, - kwargs: Any, # noqa: ARG001 + kwargs: Any, ) -> ChatCompletion: response = await wrapped(*args, **kwargs) return compute_impacts_and_return_response(response)