fixed querying

app/main.py

@@ -44,6 +44,15 @@ class QueryRequest(BaseModel):
             }
         }
 
+class CompanyQueryRequest(BaseModel):
+    question: str
+
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "question": "Find me companies in the fintech sector located in San Francisco."
+            }
+        }
 
 @app.get("/")
 def health():
@@ -120,7 +129,7 @@ async def query_investors(request: QueryRequest):
 @app.post(
     "/query-companies", response_model=PaginatedResponse[CompanyData], tags=["Querying"]
 )
-async def query_companies(request: QueryRequest):
+async def query_companies(request: CompanyQueryRequest):
     """
     Query companies using natural language.