Fix final ruff linting issues

- Remove duplicate import in __init__.py
- Clean up extra blank lines in gemini.py
- Remove extra blank line in _create_single_response method
This commit is contained in:
Pete
2025-07-20 18:58:54 -04:00
parent b1a5cddde4
commit ec361df0d1
3 changed files with 5 additions and 12 deletions

View File

@@ -1,3 +1,2 @@
from .file_api import GeminiFileAPI from .file_api import GeminiFileAPI
from .gemini import GeminiMultimodalLiveLLMService from .gemini import GeminiMultimodalLiveLLMService
from .file_api import GeminiFileAPI

View File

@@ -497,6 +497,7 @@ def parse_server_event(str):
except Exception as e: except Exception as e:
print(f"Error parsing server event: {e}") print(f"Error parsing server event: {e}")
class ContextWindowCompressionConfig(BaseModel): class ContextWindowCompressionConfig(BaseModel):
"""Configuration for context window compression. """Configuration for context window compression.

View File

@@ -61,8 +61,6 @@ from pipecat.processors.aggregators.openai_llm_context import (
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.google.frames import LLMSearchOrigin, LLMSearchResponseFrame, LLMSearchResult from pipecat.services.google.frames import LLMSearchOrigin, LLMSearchResponseFrame, LLMSearchResult
from pipecat.services.llm_service import FunctionCallFromLLM, LLMService from pipecat.services.llm_service import FunctionCallFromLLM, LLMService
from pipecat.services.openai.llm import ( from pipecat.services.openai.llm import (
OpenAIAssistantContextAggregator, OpenAIAssistantContextAggregator,
OpenAIUserContextAggregator, OpenAIUserContextAggregator,
@@ -74,11 +72,9 @@ from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_gemini_live, traced_stt from pipecat.utils.tracing.service_decorators import traced_gemini_live, traced_stt
from . import events from . import events
from .audio_transcriber import AudioTranscriber from .audio_transcriber import AudioTranscriber
from .file_api import GeminiFileAPI from .file_api import GeminiFileAPI
try: try:
import websockets import websockets
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
@@ -575,7 +571,7 @@ class GeminiMultimodalLiveLLMService(LLMService):
else {}, else {},
"extra": params.extra if isinstance(params.extra, dict) else {}, "extra": params.extra if isinstance(params.extra, dict) else {},
} }
# Initialize the File API client # Initialize the File API client
self.file_api = GeminiFileAPI(api_key=api_key, base_url=file_api_base_url) self.file_api = GeminiFileAPI(api_key=api_key, base_url=file_api_base_url)
@@ -961,7 +957,7 @@ class GeminiMultimodalLiveLLMService(LLMService):
await self._handle_evt_error(evt) await self._handle_evt_error(evt)
# errors are fatal, so exit the receive loop # errors are fatal, so exit the receive loop
return return
# #
# #
# #
@@ -1025,7 +1021,6 @@ class GeminiMultimodalLiveLLMService(LLMService):
async def _create_single_response(self, messages_list): async def _create_single_response(self, messages_list):
"""Create a single response from a list of messages.""" """Create a single response from a list of messages."""
# Refactor to combine this logic with same logic in GeminiMultimodalLiveContext # Refactor to combine this logic with same logic in GeminiMultimodalLiveContext
messages = [] messages = []
for item in messages_list: for item in messages_list:
@@ -1202,7 +1197,6 @@ class GeminiMultimodalLiveLLMService(LLMService):
self._bot_text_buffer = "" self._bot_text_buffer = ""
self._llm_output_buffer = "" self._llm_output_buffer = ""
# Process grounding metadata if we have accumulated any # Process grounding metadata if we have accumulated any
if self._accumulated_grounding_metadata: if self._accumulated_grounding_metadata:
await self._process_grounding_metadata( await self._process_grounding_metadata(
@@ -1295,7 +1289,6 @@ class GeminiMultimodalLiveLLMService(LLMService):
# Collect text for tracing # Collect text for tracing
self._llm_output_buffer += text self._llm_output_buffer += text
await self.push_frame(LLMTextFrame(text=text)) await self.push_frame(LLMTextFrame(text=text))
await self.push_frame(TTSTextFrame(text=text)) await self.push_frame(TTSTextFrame(text=text))
@@ -1335,13 +1328,13 @@ class GeminiMultimodalLiveLLMService(LLMService):
) )
chunk_to_origin[index] = origin chunk_to_origin[index] = origin
origins.append(origin) origins.append(origin)
# Add grounding support results to the appropriate origins # Add grounding support results to the appropriate origins
for support in grounding_metadata.groundingSupports: for support in grounding_metadata.groundingSupports:
if support.segment and support.groundingChunkIndices: if support.segment and support.groundingChunkIndices:
text = support.segment.text or "" text = support.segment.text or ""
confidence_scores = support.confidenceScores or [] confidence_scores = support.confidenceScores or []
# Add this result to all origins referenced by this support # Add this result to all origins referenced by this support
for chunk_index in support.groundingChunkIndices: for chunk_index in support.groundingChunkIndices:
if chunk_index in chunk_to_origin: if chunk_index in chunk_to_origin: