From be86fe0845c3d8aebda37cc9b4d3082b911a220e Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Fri, 6 Sep 2024 01:02:04 +0200
Subject: [PATCH] chore: lint

---
 semantic_router/index/pinecone.py | 7 ++++---
 semantic_router/layer.py          | 2 --
 semantic_router/llms/openai.py    | 6 ------
 3 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index 9042eafe..5199421b 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -797,7 +797,6 @@ def __len__(self):
         return self.index.describe_index_stats()["total_vector_count"]
 
 
-
 def parse_route_info(metadata: List[Dict[str, Any]]) -> List[Tuple]:
     """Parses metadata from Pinecone index to extract route, utterance, function
     schema and additional metadata.
@@ -814,12 +813,14 @@ def parse_route_info(metadata: List[Dict[str, Any]]) -> List[Tuple]:
         sr_function_schema = json.loads(record.get("sr_function_schema", "{}"))
         if sr_function_schema == {}:
             sr_function_schema = None
-        
+
         additional_metadata = {
             key: value
             for key, value in record.items()
             if key not in ["sr_route", "sr_utterance", "sr_function_schema"]
         }
         # TODO: Not a fan of tuple packing here
-        route_info.append((sr_route, sr_utterance, sr_function_schema, additional_metadata))
+        route_info.append(
+            (sr_route, sr_utterance, sr_function_schema, additional_metadata)
+        )
     return route_info
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 5991c468..42179459 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -219,7 +219,6 @@ def __init__(
         # if routes list has been passed, we initialize index now
         if self.index.sync:
             # initialize index now
-            logger.info(f"JB TEMP: {self.routes=}")
             if len(self.routes) > 0:
                 self._add_and_sync_routes(routes=self.routes)
             else:
@@ -545,7 +544,6 @@ def _add_and_sync_routes(self, routes: List[Route]):
         )
 
         # Update local route layer state
-        logger.info([data.get("function_schemas", None) for _, data in layer_routes_dict.items()])
         self.routes = []
         for route, data in layer_routes_dict.items():
             function_schemas = data.get("function_schemas", None)
diff --git a/semantic_router/llms/openai.py b/semantic_router/llms/openai.py
index 4c49974a..dfff8096 100644
--- a/semantic_router/llms/openai.py
+++ b/semantic_router/llms/openai.py
@@ -94,9 +94,6 @@ def __call__(
             tools: Union[List[Dict[str, Any]], NotGiven] = (
                 function_schemas if function_schemas else NOT_GIVEN
             )
-            logger.info(f"{function_schemas=}")
-            logger.info(f"{function_schemas is None=}")
-            logger.info(f"{tools=}")
 
             completion = self.client.chat.completions.create(
                 model=self.name,
@@ -187,8 +184,6 @@ def extract_function_inputs(
             raise Exception("No output generated for extract function input")
         output = output.replace("'", '"')
         function_inputs = json.loads(output)
-        logger.info(f"Function inputs: {function_inputs}")
-        logger.info(f"function_schemas: {function_schemas}")
         if not self._is_valid_inputs(function_inputs, function_schemas):
             raise ValueError("Invalid inputs")
         return function_inputs
@@ -206,7 +201,6 @@ async def async_extract_function_inputs(
             raise Exception("No output generated for extract function input")
         output = output.replace("'", '"')
         function_inputs = json.loads(output)
-        logger.info(f"OpenAI => Function Inputs: {function_inputs}")
         if not self._is_valid_inputs(function_inputs, function_schemas):
             raise ValueError("Invalid inputs")
         return function_inputs