nao-core 0.0.19__tar.gz → 0.0.22__tar.gz

{nao_core-0.0.19 → nao_core-0.0.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nao-core
-Version: 0.0.19
+Version: 0.0.22
 Summary: nao Core is your analytics context builder with the best chat interface.
 Project-URL: Homepage, https://getnao.io
 Project-URL: Repository, https://github.com/naolabs/chat

{nao_core-0.0.19 → nao_core-0.0.22}/nao_core/__init__.py

@@ -1,2 +1,2 @@
 # nao Core CLI package
-__version__ = "0.0.19"
+__version__ = "0.0.22"

{nao_core-0.0.19 → nao_core-0.0.22}/nao_core/bin/fastapi/main.py

@@ -1,3 +1,4 @@
+from nao_core.config import NaoConfig
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
@@ -10,8 +11,6 @@ from pathlib import Path
 cli_path = Path(__file__).parent.parent.parent / "cli"
 sys.path.insert(0, str(cli_path))
 
-from nao_core.config import NaoConfig
-
 port = int(os.environ.get("PORT", 8005))
 
 app = FastAPI()
@@ -35,23 +34,6 @@ class ExecuteSQLResponse(BaseModel):
     row_count: int
     columns: list[str]
 
-class ContextRequest(BaseModel):
-    nao_project_folder: str
-
-class ContextResponse(BaseModel):
-    context: dict
-
-@app.post("/context", response_model=ContextResponse)
-async def get_context(request: ContextRequest):
-    try:
-        # Load the nao config from the project folder
-        project_path = Path(request.nao_project_folder)
-        config = NaoConfig.try_load(project_path)
-        context = config.get_context()
-        return ContextResponse(context=context)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
 
 @app.post("/execute_sql", response_model=ExecuteSQLResponse)
 async def execute_sql(request: ExecuteSQLRequest):
@@ -102,11 +84,10 @@ async def execute_sql(request: ExecuteSQLRequest):
             )
 
         connection = db_config.connect()
-        result = connection.sql(request.sql)
-
-        df = result.to_pandas()
         
-        # Convert numpy types to native Python types for JSON serialization
+        cursor = connection.raw_sql(request.sql)  # type: ignore[attr-defined]
+        df = cursor.to_dataframe()
+        
         def convert_value(v):
             if isinstance(v, (np.integer,)):
                 return int(v)

{nao_core-0.0.19 → nao_core-0.0.22}/nao_core/commands/sync/databases.py

@@ -29,7 +29,7 @@ def sync_bigquery(
             Tuple of (datasets_synced, tables_synced)
     """
     conn = db_config.connect()
-    db_path = base_path / "bigquery" / f"database={db_config.name}"
+    db_path = base_path / "type=bigquery" / f"database={db_config.project_id}"
 
     datasets_synced = 0
     tables_synced = 0

{nao_core-0.0.19 → nao_core-0.0.22}/nao_core/config/base.py

@@ -74,12 +74,3 @@ class NaoConfig(BaseModel):
     def json_schema(cls) -> dict:
         """Generate JSON schema for the configuration."""
         return cls.model_json_schema()
-
-    def get_context(self) -> str:
-        """Get the context of the project."""
-        return f"""
-        Project name: {self.project_name}
-        Databases: {self.databases}
-        Repos: {self.repos}
-        LLM: {self.llm}
-        """

{nao_core-0.0.19 → nao_core-0.0.22}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nao-core"
-version = "0.0.19"
+version = "0.0.22"
 description = "nao Core is your analytics context builder with the best chat interface."
 readme = "README.md"
 requires-python = ">=3.10"