apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-07-02 15:51:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix hr agent

Browse source
This commit is contained in:
Adil Hafeez 2024-10-18 19:19:44 -07:00
parent b037c603cd
commit baa84eb035
12 changed files with 212 additions and 2904 deletions
Download patch file Download diff file Expand all files Collapse all files

22
demos/hr_agent/main.py
View file

@ -22,7 +22,7 @@ class RegionType(Enum):
# Define the request model
class HeadcountRequest(BaseModel):
region: RegionType
staffing_type: str
staffing_type: StaffingType
class HeadcountResponseSummary(BaseModel):
@ -32,9 +32,21 @@ class HeadcountResponseSummary(BaseModel):
HEADCOUNT = {
ASIA: {CONTRACT: 100, FTE: 150, AGENCY: 2000},
EUROPE: {CONTRACT: 80, FTE: 120, AGENCY: 2500},
AMERICAS: {CONTRACT: 90, FTE: 200, AGENCY: 3000},
RegionType.ASIA: {
StaffingType.CONTRACT: 100,
StaffingType.FTE: 150,
StaffingType.AGENCY: 2000,
},
RegionType.EUROPE: {
StaffingType.CONTRACT: 80,
StaffingType.FTE: 120,
StaffingType.AGENCY: 2500,
},
RegionType.AMERICAS: {
StaffingType.CONTRACT: 90,
StaffingType.FTE: 200,
StaffingType.AGENCY: 3000,
},
}
@ -48,7 +60,7 @@ def get_headcount(request: HeadcountRequest):
response = {
"region": request.region.value,
"staffing_type": f"Staffing agency: {staffing_type}",
"staffing_type": f"Staffing agency: {request.staffing_type}",
"headcount": f"Headcount: {headcount}",
}