Fix lint errors
This commit is contained in:
@@ -1,4 +1,8 @@
|
||||
"""RAG service for calling RAG endpoints with high concurrency."""
|
||||
|
||||
import logging
|
||||
from types import TracebackType
|
||||
from typing import Self
|
||||
|
||||
import httpx
|
||||
from pydantic import BaseModel, Field
|
||||
@@ -68,8 +72,11 @@ class RAGService:
|
||||
)
|
||||
|
||||
logger.info(
|
||||
f"RAGService initialized with endpoint: {self.rag_endpoint_url}, "
|
||||
f"max_connections: {max_connections}, timeout: {timeout}s",
|
||||
"RAGService initialized with endpoint: %s, "
|
||||
"max_connections: %s, timeout: %ss",
|
||||
self.rag_endpoint_url,
|
||||
max_connections,
|
||||
timeout,
|
||||
)
|
||||
|
||||
async def query(self, messages: list[dict[str, str]]) -> str:
|
||||
@@ -93,7 +100,7 @@ class RAGService:
|
||||
request = RAGRequest(messages=message_objects)
|
||||
|
||||
# Make async HTTP POST request
|
||||
logger.debug(f"Sending RAG request with {len(messages)} messages")
|
||||
logger.debug("Sending RAG request with %s messages", len(messages))
|
||||
|
||||
response = await self._client.post(
|
||||
self.rag_endpoint_url,
|
||||
@@ -108,32 +115,37 @@ class RAGService:
|
||||
response_data = response.json()
|
||||
rag_response = RAGResponse(**response_data)
|
||||
|
||||
logger.debug(f"RAG response received: {len(rag_response.response)} chars")
|
||||
return rag_response.response
|
||||
|
||||
logger.debug("RAG response received: %s chars", len(rag_response.response))
|
||||
except httpx.HTTPStatusError as e:
|
||||
logger.exception(
|
||||
f"HTTP error calling RAG endpoint: {e.response.status_code} - {e.response.text}",
|
||||
"HTTP error calling RAG endpoint: %s - %s",
|
||||
e.response.status_code,
|
||||
e.response.text,
|
||||
)
|
||||
raise
|
||||
except httpx.RequestError as e:
|
||||
logger.exception(f"Request error calling RAG endpoint: {e!s}")
|
||||
except httpx.RequestError:
|
||||
logger.exception("Request error calling RAG endpoint:")
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
f"Unexpected error calling RAG endpoint: {e!s}", exc_info=True,
|
||||
)
|
||||
except Exception:
|
||||
logger.exception("Unexpected error calling RAG endpoint")
|
||||
raise
|
||||
else:
|
||||
return rag_response.response
|
||||
|
||||
async def close(self) -> None:
|
||||
"""Close the HTTP client and release connections."""
|
||||
await self._client.aclose()
|
||||
logger.info("RAGService client closed")
|
||||
|
||||
async def __aenter__(self):
|
||||
async def __aenter__(self) -> Self:
|
||||
"""Async context manager entry."""
|
||||
return self
|
||||
|
||||
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||
async def __aexit__(
|
||||
self,
|
||||
exc_type: type[BaseException] | None,
|
||||
exc_val: BaseException | None,
|
||||
exc_tb: TracebackType | None,
|
||||
) -> None:
|
||||
"""Async context manager exit."""
|
||||
await self.close()
|
||||
|
||||
Reference in New Issue
Block a user