updating the demo and model-server-tests to pull from poetry

This commit is contained in:
Salman Paracha 2024-10-22 18:34:25 -07:00
parent 6a72cb45b7
commit 51259b393a
3 changed files with 16 additions and 10 deletions

View file

@ -24,7 +24,7 @@ endpoints:
# default system prompt used by all prompt targets
system_prompt: |
You are a HR agent assistant that helps HR decision makers with reporting and workfoce planning. Nothing else. Please stay on topic of HR.
You are a Workforce agent assistant that helps workforce planning and HR decision makers with reporting and workfoce planning. NOTHING ELSE. Please stay on topic of workforce planning and HR.
prompt_targets:
- name: hr_qa
@ -41,12 +41,12 @@ prompt_targets:
parameters:
- name: staffing_type
type: str
description: The staffing type like contract, fte or agency
description: Staffing type like contract, fte or agency
required: true
- name: region
type: str
required: true
description: the geographical region for which you want workforce data.
description: Geographical region for which you want workforce data like asia, europe, americas.
- name: point_in_time
type: int
required: false

View file

@ -50,7 +50,7 @@ def get_workforce(request: WorkforceRequset):
"region": region,
"staffing_type": f"Staffing agency: {staffing_type}",
"headcount": f"Headcount: {int(workforce_data_df[(workforce_data_df['region']==region) & (workforce_data_df['point_in_time']==point_in_time)][staffing_type].values[0])}",
"satisfaction": f"Satisifaction: {float(workforce_data_df[(workforce_data_df['region']==region) & (workforce_data_df['point_in_time']==point_in_time)][satisfaction].values[0])}",
"satisfaction": f"Satisifaction: {float(workforce_data_df[(workforce_data_df['region']==region) & (workforce_data_df['point_in_time']==point_in_time)]['satisfaction'].values[0])}",
}
return response