tmp

Author: hfhoffman1144

langchain-rag-app/chroma_data/5b736d9a-7fc1-4da4-ac91-503db6a8b3a8/data_level0.bin

@@ -9,12 +9,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 25,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [],
    "source": [
     "from pydantic import BaseModel, EmailStr\n",
-    "from datetime import date\n",
+    "from datetime import date, timedelta\n",
+    "from dataclasses import dataclass\n",
     "from uuid import UUID, uuid4"
    ]
   },
@@ -271,18 +272,53 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 15,
    "metadata": {},
-   "outputs": [],
-   "source": []
+   "outputs": [
+    {
+     "ename": "TypeError",
+     "evalue": "'year' is an invalid keyword argument for __new__()",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mTypeError\u001b[0m                                 Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[15], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[43mtimedelta\u001b[49m\u001b[43m(\u001b[49m\u001b[43myear\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;241;43m18\u001b[39;49m\u001b[43m)\u001b[49m\n",
+      "\u001b[0;31mTypeError\u001b[0m: 'year' is an invalid keyword argument for __new__()"
+     ]
+    }
+   ],
+   "source": [
+    "timedelta(year=18)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 16,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "1.063013698630137"
+      ]
+     },
+     "execution_count": 16,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "date_delta = date.today() - date(2023, 2, 1)\n",
+    "date_delta.days / 365"
+   ]
   },
   {
    "cell_type": "code",
    "execution_count": 65,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from pydantic import PositiveFloat, validate_call"
+    "from pydantic import PositiveFloat, validate_call, Field"
    ]
   },
   {
@@ -292,7 +328,7 @@
    "outputs": [],
    "source": [
     "@validate_call\n",
-    "def send_billing_email(client_name:str,\n",
+    "def create_billing_email(client_name:str = Field(min_length=1),\n",
     "                       client_email:EmailStr,\n",
     "                       items_purchased:list[str],\n",
     "                       amount_owed:PositiveFloat) -> None:\n",
@@ -306,9 +342,7 @@
     "    {items_purchased}\n",
     "    \"\"\"\n",
     "    \n",
-    "    print(f\"Sending to {client_email}...\")\n",
-    "    print(email_str)\n",
-    "    print(\"Sent!\")\n",
+    "    return email_str\n",
     "    "
    ]
   },

langchain-rag-app/chroma_data/5b736d9a-7fc1-4da4-ac91-503db6a8b3a8/header.bin

@@ -0,0 +1,62 @@
+from typing import Self
+from datetime import date
+from uuid import UUID, uuid4
+from enum import Enum
+from pydantic import (
+    BaseModel,
+    EmailStr,
+    Field,
+    field_validator,
+    model_validator,
+)
+
+
+class Department(Enum):
+    HR = "HR"
+    SALES = "SALES"
+    IT = "IT"
+    ENGINEERING = "ENGINEERING"
+
+
+class Employee(BaseModel):
+    employee_id: UUID = Field(default_factory=lambda: uuid4(), frozen=True)
+    name: str = Field(min_length=1, frozen=True)
+    email: EmailStr = Field(pattern=r".+@company\.com$")
+    date_of_birth: date = Field(alias="birth_date", repr=False, frozen=True)
+    salary: float = Field(alias="compensation", gt=0, repr=False)
+    department: Department
+    elected_benefits: bool
+
+    @field_validator("date_of_birth")
+    @classmethod
+    def check_valid_age(cls, date_of_birth: date) -> date:
+
+        date_delta = date.today() - date_of_birth
+        age = date_delta.days / 365
+
+        if age < 18:
+            raise ValueError("Employees must be at least 18 years old.")
+
+        return date_of_birth
+
+    @model_validator(mode="after")
+    def check_it_benefits(self) -> Self:
+
+        department = self.department
+        elected_benefits = self.elected_benefits
+
+        if department == Department.IT and elected_benefits:
+            raise ValueError(
+                "IT employees are contractors and don't qualify for benefits."
+            )
+        return self
+
+
+new_employee = {
+    "name": "Chris DeTuma",
+    "email": "[email protected]",
+    "birth_date": "1998-04-02",
+    "compensation": 100_000,
+    "department": "IT",
+    "elected_benefits": True,
+}