lint + formating with black (#158)

* lint + formating with black

* add black as pre commit

demos/insurance_agent/insurance_agent_main.py

@@ -3,28 +3,40 @@ from pydantic import BaseModel, Field
 
 app = FastAPI()
 
+
 class Conversation(BaseModel):
     arch_messages: list
 
+
 class PolicyCoverageRequest(BaseModel):
-    policy_type: str = Field(..., description="The type of a policy held by the customer For, e.g. car, boat, house, motorcycle)")
+    policy_type: str = Field(
+        ...,
+        description="The type of a policy held by the customer For, e.g. car, boat, house, motorcycle)",
+    )
+
 
 class PolicyInitiateRequest(PolicyCoverageRequest):
-    deductible: float = Field(..., description="The deductible amount set of the policy")
+    deductible: float = Field(
+        ..., description="The deductible amount set of the policy"
+    )
+
 
 class ClaimUpdate(BaseModel):
     claim_id: str
     notes: str  # Status or details of the claim
 
+
 class DeductibleUpdate(BaseModel):
     policy_id: str
     deductible: float
 
+
 class CoverageResponse(BaseModel):
     policy_type: str
     coverage: str  # Description of coverage
     premium: float  # The premium cost
 
+
 # Get information about policy coverage
 @app.post("/policy/coverage", response_model=CoverageResponse)
 async def get_policy_coverage(req: PolicyCoverageRequest):
@@ -32,10 +44,22 @@ async def get_policy_coverage(req: PolicyCoverageRequest):
     Retrieve the coverage details for a given policy type (car, boat, house, motorcycle).
     """
     policy_coverage = {
-        "car": {"coverage": "Full car coverage with collision, liability", "premium": 500.0},
-        "boat": {"coverage": "Full boat coverage including theft and storm damage", "premium": 700.0},
-        "house": {"coverage": "Full house coverage including fire, theft, flood", "premium": 1000.0},
-        "motorcycle": {"coverage": "Full motorcycle coverage with liability", "premium": 400.0},
+        "car": {
+            "coverage": "Full car coverage with collision, liability",
+            "premium": 500.0,
+        },
+        "boat": {
+            "coverage": "Full boat coverage including theft and storm damage",
+            "premium": 700.0,
+        },
+        "house": {
+            "coverage": "Full house coverage including fire, theft, flood",
+            "premium": 1000.0,
+        },
+        "motorcycle": {
+            "coverage": "Full motorcycle coverage with liability",
+            "premium": 400.0,
+        },
     }
 
     if req.policy_type not in policy_coverage:
@@ -44,9 +68,10 @@ async def get_policy_coverage(req: PolicyCoverageRequest):
     return CoverageResponse(
         policy_type=req.policy_type,
         coverage=policy_coverage[req.policy_type]["coverage"],
-        premium=policy_coverage[req.policy_type]["premium"]
+        premium=policy_coverage[req.policy_type]["premium"],
     )
 
+
 # Initiate policy coverage
 @app.post("/policy/initiate")
 async def initiate_policy(policy_request: PolicyInitiateRequest):
@@ -56,7 +81,11 @@ async def initiate_policy(policy_request: PolicyInitiateRequest):
     if policy_request.policy_type not in ["car", "boat", "house", "motorcycle"]:
         raise HTTPException(status_code=400, detail="Invalid policy type")
 
-    return {"message": f"Policy initiated for {policy_request.policy_type}", "deductible": policy_request.deductible}
+    return {
+        "message": f"Policy initiated for {policy_request.policy_type}",
+        "deductible": policy_request.deductible,
+    }
+
 
 # Update claim details
 @app.post("/policy/claim")
@@ -65,8 +94,11 @@ async def update_claim(req: ClaimUpdate):
     Update the status or details of a claim.
     """
     # For simplicity, this is a mock update response
-    return {"message": f"Claim {claim_update.claim_id} for policy {claim_update.claim_id} has been updated",
-            "update": claim_update.notes}
+    return {
+        "message": f"Claim {claim_update.claim_id} for policy {claim_update.claim_id} has been updated",
+        "update": claim_update.notes,
+    }
+
 
 # Update deductible amount
 @app.post("/policy/deductible")
@@ -75,8 +107,11 @@ async def update_deductible(deductible_update: DeductibleUpdate):
     Update the deductible amount for a specific policy.
     """
     # For simplicity, this is a mock update response
-    return {"message": f"Deductible for policy {deductible_update.policy_id} has been updated",
-            "new_deductible": deductible_update.deductible}
+    return {
+        "message": f"Deductible for policy {deductible_update.policy_id} has been updated",
+        "new_deductible": deductible_update.deductible,
+    }
+
 
 # Post method for policy Q/A
 @app.post("/policy/qa")
@@ -86,21 +121,20 @@ async def policy_qa(conversation: Conversation):
     It forwards the conversation to the OpenAI client via a local proxy and returns the response.
     """
     return {
-            "choices": [
-              {
+        "choices": [
+            {
                 "message": {
-                  "role": "assistant",
-                  "content": "I am a helpful insurance agent, and can only help with insurance things"
+                    "role": "assistant",
+                    "content": "I am a helpful insurance agent, and can only help with insurance things",
                 },
                 "finish_reason": "completed",
-                "index": 0
-              }
-            ],
-            "model": "insurance_agent",
-            "usage": {
-              "completion_tokens": 0
+                "index": 0,
             }
-          }
+        ],
+        "model": "insurance_agent",
+        "usage": {"completion_tokens": 0},
+    }
+
 
 # Run the app using:
 # uvicorn main:app --reload