debug errors

Author: yrobla

Makefile

@@ -21,7 +21,7 @@ lint:
 	poetry run ruff check .
 
 test:
-	poetry run pytest
+	poetry run pytest -s
 
 security:
 	poetry run bandit -r src/

src/codegate/cli.py

@@ -120,6 +120,7 @@ def serve(
     """Start the codegate server."""
     logger = None
     try:
+        click.echo("In serve")
         # Create provider URLs dict from CLI options
         cli_provider_urls: Dict[str, str] = {}
         if vllm_url:
@@ -166,12 +167,16 @@ def serve(
         )
 
     except KeyboardInterrupt:
+        click.echo("in keyboard")
         if logger:
             logger.info("Shutting down server")
     except ConfigurationError as e:
+        click.echo("in config error")
         click.echo(f"Configuration error: {e}", err=True)
         sys.exit(1)
     except Exception as e:
+        click.echo("exception is")
+        click.echo(e)
         if logger:
             logger.exception("Unexpected error occurred")
         click.echo(f"Error: {e}", err=True)

src/codegate/storage/storage_engine.py

@@ -39,13 +39,22 @@ def __init__(self, data_path='./weaviate_data'):
         self.schema_config = schema_config
 
         # setup schema for weaviate
+        print("before get client")
         weaviate_client = self.get_client(self.data_path)
+        print("after get client")
         if weaviate_client is not None:
             try:
+                print("before connect")
                 weaviate_client.connect()
+                print("before setup schema")
                 self.setup_schema(weaviate_client)
+                print("after setup schema")
+            except Exception as e:
+                self.__logger.error(f"Failed to connect or setup schema: {str(e)}")                
             finally:
+                print("before close")
                 weaviate_client.close()
+                print("after close")
         else:
             self.__logger.error("Could not find client, skipping schema setup.")