fast-agent-mcp 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

mcp_agent/resources/examples/researcher/researcher-imp.py

@@ -1,4 +1,5 @@
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 agents = FastAgent(name="Enhanced Researcher")
@@ -162,7 +163,7 @@ The researcher should be able to understand exactly why they received their rati
     min_rating="EXCELLENT",
     name="EnhancedResearcher",
 )
-async def main():
+async def main() -> None:
     async with agents.run() as agent:
         # Start with a warm-up to set expectations and explain the research approach
         await agent.Researcher.send(
@@ -180,9 +181,7 @@ async def main():
         # Start the main research workflow
         await agent.prompt("EnhancedResearcher")
 
-        print(
-            "\nWould you like to ask follow-up questions to the Researcher? (Type 'STOP' to end)"
-        )
+        print("\nWould you like to ask follow-up questions to the Researcher? (Type 'STOP' to end)")
         await agent.prompt("Researcher", default="STOP")
 
 

mcp_agent/resources/examples/researcher/researcher.py

@@ -1,6 +1,7 @@
 import asyncio
 
 from mcp_agent.core.fastagent import FastAgent
+
 # from rich import print
 
 agents = FastAgent(name="Researcher Agent")
@@ -16,7 +17,7 @@ The interpreter has numpy, pandas, matplotlib and seaborn already installed
     """,
     servers=["brave", "interpreter", "filesystem", "fetch"],
 )
-async def main():
+async def main() -> None:
     research_prompt = """
 Produce an investment report for the company Eutelsat. The final report should be saved in the filesystem in markdown format, and
 contain at least the following: 

mcp_agent/resources/examples/workflows/agent_build.py

@@ -3,6 +3,7 @@ This demonstrates creating multiple agents and an orchestrator to coordinate the
 """
 
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 from mcp_agent.workflows.llm.augmented_llm import RequestParams
 
@@ -62,7 +63,7 @@ if needed. Remind the Human of this.
     request_params=RequestParams(maxTokens=8192),
     max_iterations=5,
 )
-async def main():
+async def main() -> None:
     async with fast.run() as agent:
         CODER_WARMUP = """
 - Read this paper: https://www.anthropic.com/research/building-effective-agents" to understand how 

mcp_agent/resources/examples/workflows/chaining.py

@@ -1,4 +1,5 @@
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 # Create the application
@@ -21,7 +22,7 @@ fast = FastAgent("Agent Chaining")
     name="post_writer",
     sequence=["url_fetcher", "social_media"],
 )
-async def main():
+async def main() -> None:
     async with fast.run() as agent:
         # using chain workflow
         await agent.post_writer.prompt()

mcp_agent/resources/examples/workflows/evaluator.py

@@ -3,6 +3,7 @@ This demonstrates creating an optimizer and evaluator to iteratively improve con
 """
 
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 # Create the application
@@ -49,7 +50,7 @@ fast = FastAgent("Evaluator-Optimizer")
     min_rating="EXCELLENT",  # Strive for excellence
     max_refinements=3,  # Maximum iterations
 )
-async def main():
+async def main() -> None:
     async with fast.run() as agent:
         job_posting = (
             "Software Engineer at LastMile AI. Responsibilities include developing AI systems, "

mcp_agent/resources/examples/workflows/human_input.py

@@ -3,6 +3,7 @@ Agent which demonstrates Human Input tool
 """
 
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 # Create the application
@@ -14,7 +15,7 @@ fast = FastAgent("Human Input")
     instruction="An AI agent that assists with basic tasks. Request Human Input when needed.",
     human_input=True,
 )
-async def main():
+async def main() -> None:
     async with fast.run() as agent:
         # this usually causes the LLM to request the Human Input Tool
         await agent("print the next number in the sequence")

mcp_agent/resources/examples/workflows/orchestrator.py

@@ -3,6 +3,7 @@ This demonstrates creating multiple agents and an orchestrator to coordinate the
 """
 
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 # Create the application
@@ -47,7 +48,7 @@ fast = FastAgent("Orchestrator-Workers")
     agents=["finder", "writer", "proofreader"],
     plan_type="iterative",
 )
-async def main():
+async def main() -> None:
     async with fast.run() as agent:
         await agent()
         await agent.author(

mcp_agent/resources/examples/workflows/parallel.py

@@ -3,6 +3,7 @@ Parallel Workflow showing Fan Out and Fan In agents, using different models
 """
 
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 # Create the application
@@ -65,7 +66,7 @@ and whispers of a hidden agenda linger among the villagers.
     fan_in="grader",
     name="parallel",
 )
-async def main():
+async def main() -> None:
     # Use the app's context manager
     async with fast.run() as agent:
         await agent.parallel(f"student short story submission: {SHORT_STORY}")

mcp_agent/resources/examples/workflows/router.py

@@ -6,6 +6,7 @@ Demonstrates router's ability to either:
 """
 
 import asyncio
+
 from mcp_agent.core.fastagent import FastAgent
 
 # Create the application
@@ -43,7 +44,7 @@ SAMPLE_REQUESTS = [
     model="sonnet",
     agents=["code_expert", "general_assistant", "fetcher"],
 )
-async def main():
+async def main() -> None:
     async with fast.run() as agent:
         for request in SAMPLE_REQUESTS:
             await agent.route(request)

mcp_agent/resources/examples/workflows/sse.py

@@ -10,7 +10,7 @@ fa = FastAgent("My Application")
 @fa.agent("analyst", "hello, world", servers=["fetch"])
 
 # Run the application with MCP server
-async def main():
+async def main() -> None:
     await fa.run_with_mcp_server(
         transport="sse",  # Use "sse" for web server, "stdio" for command line
         port=8000,

mcp_agent/telemetry/usage_tracking.py

@@ -1,10 +1,11 @@
 import logging
+
 from mcp_agent.config import get_settings
 
 logger = logging.getLogger(__name__)
 
 
-def send_usage_data():
+def send_usage_data() -> None:
     config = get_settings()
     if not config.usage_telemetry.enabled:
         logger.info("Usage tracking is disabled")

mcp_agent/ui/console_display.py

@@ -17,7 +17,7 @@ class ConsoleDisplay:
     This centralizes the UI display logic used by LLM implementations.
     """
 
-    def __init__(self, config=None):
+    def __init__(self, config=None) -> None:
         """
         Initialize the console display handler.
 
@@ -26,7 +26,7 @@ class ConsoleDisplay:
         """
         self.config = config
 
-    def show_tool_result(self, result: CallToolResult):
+    def show_tool_result(self, result: CallToolResult) -> None:
         """Display a tool result in a formatted panel."""
         if not self.config or not self.config.logger.show_tools:
             return
@@ -49,7 +49,7 @@ class ConsoleDisplay:
         console.console.print(panel)
         console.console.print("\n")
 
-    def show_oai_tool_result(self, result):
+    def show_oai_tool_result(self, result) -> None:
         """Display an OpenAI tool result in a formatted panel."""
         if not self.config or not self.config.logger.show_tools:
             return
@@ -70,7 +70,7 @@ class ConsoleDisplay:
         console.console.print(panel)
         console.console.print("\n")
 
-    def show_tool_call(self, available_tools, tool_name, tool_args):
+    def show_tool_call(self, available_tools, tool_name, tool_args) -> None:
         """Display a tool call in a formatted panel."""
         if not self.config or not self.config.logger.show_tools:
             return
@@ -109,25 +109,13 @@ class ConsoleDisplay:
                     tool_call_name = display_tool["name"]
             else:
                 # Handle potential object format (e.g., Pydantic models)
-                tool_call_name = (
-                    display_tool.function.name
-                    if hasattr(display_tool, "function")
-                    else display_tool.name
-                )
-
-            parts = (
-                tool_call_name.split(SEP)
-                if SEP in tool_call_name
-                else [tool_call_name, tool_call_name]
-            )
+                tool_call_name = display_tool.function.name if hasattr(display_tool, "function") else display_tool.name
+
+            parts = tool_call_name.split(SEP) if SEP in tool_call_name else [tool_call_name, tool_call_name]
 
             if selected_tool_name.split(SEP)[0] == parts[0]:
-                style = (
-                    "magenta" if tool_call_name == selected_tool_name else "dim white"
-                )
-                shortened_name = (
-                    parts[1] if len(parts[1]) <= 12 else parts[1][:11] + "…"
-                )
+                style = "magenta" if tool_call_name == selected_tool_name else "dim white"
+                shortened_name = parts[1] if len(parts[1]) <= 12 else parts[1][:11] + "…"
                 display_tool_list.append(f"[{shortened_name}] ", style)
 
         return display_tool_list
@@ -139,7 +127,7 @@ class ConsoleDisplay:
         highlight_namespaced_tool: str = "",
         title: str = "ASSISTANT",
         name: Optional[str] = None,
-    ):
+    ) -> None:
         """Display an assistant message in a formatted panel."""
         if not self.config or not self.config.logger.show_chat:
             return
@@ -150,19 +138,11 @@ class ConsoleDisplay:
             # Add human input tool if available
             tools = await aggregator.list_tools()
             if any(tool.name == HUMAN_INPUT_TOOL_NAME for tool in tools.tools):
-                style = (
-                    "green"
-                    if highlight_namespaced_tool == HUMAN_INPUT_TOOL_NAME
-                    else "dim white"
-                )
+                style = "green" if highlight_namespaced_tool == HUMAN_INPUT_TOOL_NAME else "dim white"
                 display_server_list.append("[human] ", style)
 
             # Add all available servers
-            mcp_server_name = (
-                highlight_namespaced_tool.split(SEP)[0]
-                if SEP in highlight_namespaced_tool
-                else highlight_namespaced_tool
-            )
+            mcp_server_name = highlight_namespaced_tool.split(SEP)[0] if SEP in highlight_namespaced_tool else highlight_namespaced_tool
 
             for server_name in await aggregator.list_servers():
                 style = "green" if server_name == mcp_server_name else "dim white"
@@ -181,9 +161,7 @@ class ConsoleDisplay:
         console.console.print(panel)
         console.console.print("\n")
 
-    def show_user_message(
-        self, message, model: Optional[str], chat_turn: int, name: Optional[str] = None
-    ):
+    def show_user_message(self, message, model: Optional[str], chat_turn: int, name: Optional[str] = None) -> None:
         """Display a user message in a formatted panel."""
         if not self.config or not self.config.logger.show_chat:
             return
@@ -209,7 +187,7 @@ class ConsoleDisplay:
         agent_name: Optional[str] = None,
         aggregator=None,
         arguments: Optional[dict[str, str]] = None,
-    ):
+    ) -> None:
         """
         Display information about a loaded prompt template.
 
@@ -242,9 +220,7 @@ class ConsoleDisplay:
 
         # Create content text
         content = Text()
-        messages_phrase = (
-            f"Loaded {message_count} message{'s' if message_count != 1 else ''}"
-        )
+        messages_phrase = f"Loaded {message_count} message{'s' if message_count != 1 else ''}"
         content.append(f"{messages_phrase} from template ", style="cyan italic")
         content.append(f"'{prompt_name}'", style="cyan bold italic")
 

mcp_agent/workflows/embedding/embedding_base.py

@@ -7,7 +7,6 @@ from sklearn.metrics.pairwise import cosine_similarity
 
 from mcp_agent.context_dependent import ContextDependent
 
-
 FloatArray = NDArray[float32]
 
 
@@ -32,9 +31,7 @@ class EmbeddingModel(ABC, ContextDependent):
         """Return the dimensionality of the embeddings"""
 
 
-def compute_similarity_scores(
-    embedding_a: FloatArray, embedding_b: FloatArray
-) -> Dict[str, float]:
+def compute_similarity_scores(embedding_a: FloatArray, embedding_b: FloatArray) -> Dict[str, float]:
     """
     Compute different similarity metrics between embeddings
     """

mcp_agent/workflows/embedding/embedding_cohere.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional
 
 from cohere import Client
 from numpy import array, float32
@@ -17,7 +17,7 @@ class CohereEmbeddingModel(EmbeddingModel):
         model: str = "embed-multilingual-v3.0",
         context: Optional["Context"] = None,
         **kwargs,
-    ):
+    ) -> None:
         super().__init__(context=context, **kwargs)
         self.client = Client(api_key=self.context.config.cohere.api_key)
         self.model = model

mcp_agent/workflows/embedding/embedding_openai.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional
 
 from numpy import array, float32, stack
 from openai import OpenAI
@@ -12,9 +12,7 @@ if TYPE_CHECKING:
 class OpenAIEmbeddingModel(EmbeddingModel):
     """OpenAI embedding model implementation"""
 
-    def __init__(
-        self, model: str = "text-embedding-3-small", context: Optional["Context"] = None
-    ):
+    def __init__(self, model: str = "text-embedding-3-small", context: Optional["Context"] = None) -> None:
         super().__init__(context=context)
         self.client = OpenAI(api_key=self.context.config.openai.api_key)
         self.model = model
@@ -25,20 +23,13 @@ class OpenAIEmbeddingModel(EmbeddingModel):
         }[model]
 
     async def embed(self, data: List[str]) -> FloatArray:
-        response = self.client.embeddings.create(
-            model=self.model, input=data, encoding_format="float"
-        )
+        response = self.client.embeddings.create(model=self.model, input=data, encoding_format="float")
 
         # Sort the embeddings by their index to ensure correct order
         sorted_embeddings = sorted(response.data, key=lambda x: x["index"])
 
         # Stack all embeddings into a single array
-        embeddings = stack(
-            [
-                array(embedding["embedding"], dtype=float32)
-                for embedding in sorted_embeddings
-            ]
-        )
+        embeddings = stack([array(embedding["embedding"], dtype=float32) for embedding in sorted_embeddings])
         return embeddings
 
     @property

mcp_agent/workflows/evaluator_optimizer/evaluator_optimizer.py

@@ -1,8 +1,12 @@
 import contextlib
 from enum import Enum
-from typing import Callable, List, Optional, Type, TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable, List, Optional, Type
+
 from pydantic import BaseModel, Field
 
+from mcp_agent.agents.agent import Agent
+from mcp_agent.core.agent_types import AgentConfig
+from mcp_agent.logging.logger import get_logger
 from mcp_agent.workflows.llm.augmented_llm import (
     AugmentedLLM,
     MessageParamT,
@@ -10,9 +14,6 @@ from mcp_agent.workflows.llm.augmented_llm import (
     ModelT,
     RequestParams,
 )
-from mcp_agent.agents.agent import Agent
-from mcp_agent.core.agent_types import AgentConfig
-from mcp_agent.logging.logger import get_logger
 from mcp_agent.workflows.llm.augmented_llm_passthrough import PassthroughLLM
 
 if TYPE_CHECKING:
@@ -34,15 +35,9 @@ class EvaluationResult(BaseModel):
     """Model representing the evaluation result from the evaluator LLM"""
 
     rating: QualityRating = Field(description="Quality rating of the response")
-    feedback: str = Field(
-        description="Specific feedback and suggestions for improvement"
-    )
-    needs_improvement: bool = Field(
-        description="Whether the output needs further improvement"
-    )
-    focus_areas: List[str] = Field(
-        default_factory=list, description="Specific areas to focus on in next iteration"
-    )
+    feedback: str = Field(description="Specific feedback and suggestions for improvement")
+    needs_improvement: bool = Field(description="Whether the output needs further improvement")
+    focus_areas: List[str] = Field(default_factory=list, description="Specific areas to focus on in next iteration")
 
 
 class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
@@ -75,7 +70,7 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
             use_history=self.generator_use_history,  # Use generator's history setting
         )
 
-    def _init_request_params(self):
+    def _init_request_params(self) -> None:
         """Initialize request parameters for both generator and evaluator components."""
         # Set up workflow's default params based on generator's history setting
         self.default_request_params = self._initialize_default_params({})
@@ -94,7 +89,7 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
         context: Optional["Context"] = None,
         name: Optional[str] = None,
         instruction: Optional[str] = None,
-    ):
+    ) -> None:
         """
         Initialize the evaluator-optimizer workflow.
 
@@ -120,14 +115,10 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
         if isinstance(generator, Agent):
             self.generator_use_history = generator.config.use_history
         elif isinstance(generator, AugmentedLLM):
-            if hasattr(generator, "aggregator") and isinstance(
-                generator.aggregator, Agent
-            ):
+            if hasattr(generator, "aggregator") and isinstance(generator.aggregator, Agent):
                 self.generator_use_history = generator.aggregator.config.use_history
             elif hasattr(generator, "default_request_params"):
-                self.generator_use_history = getattr(
-                    generator.default_request_params, "use_history", False
-                )
+                self.generator_use_history = getattr(generator.default_request_params, "use_history", False)
         # All other types default to False
 
         # Initialize parent class
@@ -140,20 +131,12 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
         # Set up the generator based on type
         if isinstance(generator, Agent):
             if not llm_factory:
-                raise ValueError(
-                    "llm_factory is required when using an Agent generator"
-                )
+                raise ValueError("llm_factory is required when using an Agent generator")
 
             # Use existing LLM if available, otherwise create new one
-            self.generator_llm = getattr(generator, "_llm", None) or llm_factory(
-                agent=generator
-            )
+            self.generator_llm = getattr(generator, "_llm", None) or llm_factory(agent=generator)
             self.aggregator = generator
-            self.instruction = instruction or (
-                generator.instruction
-                if isinstance(generator.instruction, str)
-                else None
-            )
+            self.instruction = instruction or (generator.instruction if isinstance(generator.instruction, str) else None)
         elif isinstance(generator, AugmentedLLM):
             self.generator_llm = generator
             self.aggregator = getattr(generator, "aggregator", None)
@@ -162,16 +145,12 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
             # ChainProxy-like object
             self.generator_llm = generator
             self.aggregator = None
-            self.instruction = (
-                instruction or f"Chain of agents: {', '.join(generator._sequence)}"
-            )
+            self.instruction = instruction or f"Chain of agents: {', '.join(generator._sequence)}"
 
         # Set up the evaluator - always disable history
         if isinstance(evaluator, str):
             if not llm_factory:
-                raise ValueError(
-                    "llm_factory is required when using a string evaluator"
-                )
+                raise ValueError("llm_factory is required when using a string evaluator")
 
             evaluator_agent = Agent(
                 name="Evaluator",
@@ -186,15 +165,11 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
             self.evaluator_llm = llm_factory(agent=evaluator_agent)
         elif isinstance(evaluator, Agent):
             if not llm_factory:
-                raise ValueError(
-                    "llm_factory is required when using an Agent evaluator"
-                )
+                raise ValueError("llm_factory is required when using an Agent evaluator")
 
             # Disable history and use/create LLM
             evaluator.config.use_history = False
-            self.evaluator_llm = getattr(evaluator, "_llm", None) or llm_factory(
-                agent=evaluator
-            )
+            self.evaluator_llm = getattr(evaluator, "_llm", None) or llm_factory(agent=evaluator)
         elif isinstance(evaluator, AugmentedLLM):
             self.evaluator_llm = evaluator
             # Ensure history is disabled
@@ -282,10 +257,7 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
                     )
 
                 # Check if we've reached acceptable quality
-                if (
-                    evaluation_result.rating.value >= self.min_rating.value
-                    or not evaluation_result.needs_improvement
-                ):
+                if evaluation_result.rating.value >= self.min_rating.value or not evaluation_result.needs_improvement:
                     logger.debug(
                         f"Acceptable quality {evaluation_result.rating.value} reached",
                         data={
@@ -338,9 +310,7 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
         request_params: RequestParams | None = None,
     ) -> ModelT:
         """Generate an optimized structured response"""
-        response_str = await self.generate_str(
-            message=message, request_params=request_params
-        )
+        response_str = await self.generate_str(message=message, request_params=request_params)
 
         return await self.generator.generate_structured(
             message=response_str,
@@ -348,9 +318,7 @@ class EvaluatorOptimizerLLM(AugmentedLLM[MessageParamT, MessageT]):
             request_params=request_params,
         )
 
-    def _build_eval_prompt(
-        self, original_request: str, current_response: str, iteration: int
-    ) -> str:
+    def _build_eval_prompt(self, original_request: str, current_response: str, iteration: int) -> str:
         """Build the evaluation prompt for the evaluator"""
         return f"""
 You are an expert evaluator for content quality. Your task is to evaluate a response against the user's original request.
@@ -409,9 +377,7 @@ Be concrete and actionable in your recommendations.
         """Build the refinement prompt for the optimizer"""
         # Get the correct history setting - use param if provided, otherwise class default
         if use_history is None:
-            use_history = (
-                self.generator_use_history
-            )  # Use generator's setting as default
+            use_history = self.generator_use_history  # Use generator's setting as default
 
         # Start with clear non-delimited instructions
         prompt = f"""

mcp_agent/workflows/intent_classifier/intent_classifier_base.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
-from typing import Dict, List, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, Dict, List, Optional
+
 from pydantic import BaseModel, Field
 
 if TYPE_CHECKING:
@@ -54,9 +55,7 @@ class IntentClassifier(ABC):
         - Determining the type of analysis requested for a dataset
     """
 
-    def __init__(
-        self, intents: List[Intent], context: Optional["Context"] = None, **kwargs
-    ):
+    def __init__(self, intents: List[Intent], context: Optional["Context"] = None, **kwargs) -> None:
         super().__init__(context=context, **kwargs)
         self.intents = {intent.name: intent for intent in intents}
         self.initialized: bool = False
@@ -65,9 +64,7 @@ class IntentClassifier(ABC):
             raise ValueError("At least one intent must be provided")
 
     @abstractmethod
-    async def classify(
-        self, request: str, top_k: int = 1
-    ) -> List[IntentClassificationResult]:
+    async def classify(self, request: str, top_k: int = 1) -> List[IntentClassificationResult]:
         """
         Classify the input request into one or more intents.
 
@@ -79,7 +76,7 @@ class IntentClassifier(ABC):
             List of classification results, ordered by confidence
         """
 
-    async def initialize(self):
+    async def initialize(self) -> None:
         """Initialize the classifier. Override this method if needed."""
         self.initialized = True
 

mcp_agent/workflows/intent_classifier/intent_classifier_embedding.py

@@ -1,17 +1,17 @@
-from typing import List, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional
 
 from numpy import mean
 
 from mcp_agent.workflows.embedding.embedding_base import (
-    FloatArray,
     EmbeddingModel,
+    FloatArray,
     compute_confidence,
     compute_similarity_scores,
 )
 from mcp_agent.workflows.intent_classifier.intent_classifier_base import (
     Intent,
-    IntentClassifier,
     IntentClassificationResult,
+    IntentClassifier,
 )
 
 if TYPE_CHECKING:
@@ -43,7 +43,7 @@ class EmbeddingIntentClassifier(IntentClassifier):
         embedding_model: EmbeddingModel,
         context: Optional["Context"] = None,
         **kwargs,
-    ):
+    ) -> None:
         super().__init__(intents=intents, context=context, **kwargs)
         self.embedding_model = embedding_model
         self.initialized = False
@@ -65,7 +65,7 @@ class EmbeddingIntentClassifier(IntentClassifier):
         await instance.initialize()
         return instance
 
-    async def initialize(self):
+    async def initialize(self) -> None:
         """
         Precompute embeddings for all intents by combining their
         descriptions and examples
@@ -91,9 +91,7 @@ class EmbeddingIntentClassifier(IntentClassifier):
 
         self.initialized = True
 
-    async def classify(
-        self, request: str, top_k: int = 1
-    ) -> List[IntentClassificationResult]:
+    async def classify(self, request: str, top_k: int = 1) -> List[IntentClassificationResult]:
         """
         Classify the input text into one or more intents
 
@@ -116,9 +114,7 @@ class EmbeddingIntentClassifier(IntentClassifier):
             if intent.embedding is None:
                 continue
 
-            similarity_scores = compute_similarity_scores(
-                request_embedding, intent.embedding
-            )
+            similarity_scores = compute_similarity_scores(request_embedding, intent.embedding)
 
             # Compute overall confidence score
             confidence = compute_confidence(similarity_scores)