Add get_interviewer_teams_in_time_slot tool

app.py

@@ -64,10 +64,10 @@ def get_interviewer_teams_by_skills(
     """
     A function to get all possible teams of interviewers by their skills.
     Args:
-        team_size: int - The number of interviewers in each team. Required parameter.
-        division: str - A division interviewers belong to. Required parameter.
-        interview_level: str - A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on. Required parameter.
-        skills: list - Skills that an interviewers must have to be able to conduct an interview. Optional parameter.
+        team_size: int - The number of interviewers in each team.
+        division: str - A division interviewers belong to.
+        interview_level: str - A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.
+        skills: list - Skills that an interviewers must have to be able to conduct an interview.
 
     Returns: 
         str: A list of interviewers teams in json.
@@ -80,6 +80,32 @@ def get_interviewer_teams_by_skills(
         duration_hours=0, 
         skills=skills)
 
+def get_interviewer_teams_in_time_slot(
+        team_size: int,
+        division: str,
+        interview_level: str,
+        start_date_time: str, 
+        duration_hours: int) -> str:
+    """
+    A function to get all possible teams of interviewers in requested time slot.
+    Args:
+        team_size: int - The number of interviewers in each team.
+        division: str - A division interviewers belong to.
+        interview_level: str - A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.
+        start_date_time: str - Date and time of start of interview.
+        duration_hours: int - Required duration of the availability in hours.
+
+    Returns: 
+        str: A list of interviewers teams in json.
+    """
+    return get_interviewer_teams_by_skills_in_time_slot(
+        team_size=team_size, 
+        division=division, 
+        interview_level=interview_level, 
+        start_date_time=start_date_time,
+        duration_hours=duration_hours, 
+        skills=[])
+
 def get_interviewer_teams_by_skills_in_time_slot(
         team_size: int,
         division: str,
@@ -90,12 +116,12 @@ def get_interviewer_teams_by_skills_in_time_slot(
     """
     A function to get all possible teams of interviewers by their skills in requested time slot.
     Args:
-        team_size: int - The number of interviewers in each team. Required parameter.
-        division: str - A division interviewers belong to. Required parameter.
-        interview_level: str - A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on. Required parameter.
-        start_date_time: str - Date and time of start of interview. Optional parameter.
-        duration_hours: int - Required duration of the availability in hours. Default value is 11.
-        skills: list - Skills that an interviewers must have to be able to conduct an interview. Optional parameter.
+        team_size: int - The number of interviewers in each team.
+        division: str - A division interviewers belong to.
+        interview_level: str - A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.
+        start_date_time: str - Date and time of start of interview.
+        duration_hours: int - Required duration of the availability in hours.
+        skills: list - Skills that an interviewers must have to be able to conduct an interview.
 
     Returns: 
         str: A list of interviewers teams in json.
@@ -115,7 +141,7 @@ def get_interviewer_teams_by_skills_in_time_slot(
         params += optional_params
 
     params += "".join(f'&skills={skill}' for skill in skills)
-    print(params)        
+    print(params)    
     
     basic_url = "http://18.133.247.78/search/interviewerTeams"    
     url = basic_url + params
@@ -130,29 +156,39 @@ class GetInterviewerTeams(BaseModel):
     """
     Pydantic arguments schema for get_interviewer_teams function
     """
-    team_size: int = Field(..., description="The number of interviewers in each team. Required parameter.")
-    division: str = Field(..., description="A division interviewers belong to. Required parameter.")
-    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on. Required parameter.")  
+    team_size: int = Field(..., description="The number of interviewers in each team.")
+    division: str = Field(..., description="A division interviewers belong to.")
+    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.")  
 
 class GetInterviewerTeamsBySkills(BaseModel):
     """
     Pydantic arguments schema for get_interviewer_teams_by_skills function
     """
-    team_size: int = Field(..., description="The number of interviewers in each team. Required parameter.")
-    division: str = Field(..., description="A division interviewers belong to. Required parameter.")
-    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on. Required parameter.")
-    skills: list = Field(..., description="Skills that an interviewers must have to be able to conduct an interview. Optional parameter.")
+    team_size: int = Field(..., description="The number of interviewers in each team.")
+    division: str = Field(..., description="A division interviewers belong to.")
+    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.")
+    skills: list = Field(..., description="Skills that an interviewers must have to be able to conduct an interview.")
+
+class GetInterviewerTeamsInTimeSlot(BaseModel):
+    """
+    Pydantic arguments schema for get_interviewer_teams_in_time_slot function
+    """
+    team_size: int = Field(..., description="The number of interviewers in each team.")
+    division: str = Field(..., description="A division interviewers belong to.")
+    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.")
+    start_date_time: str = Field(..., description="Date and time of start of interview.")
+    duration_hours: int = Field(..., description="Required duration of the availability in hours.")   
 
 class GetInterviewerTeamsBySkillsInTimeSlot(BaseModel):
     """
     Pydantic arguments schema for get_interviewer_teams_by_skills_in_time_slot function
     """
-    team_size: int = Field(..., description="The number of interviewers in each team. Required parameter.")
-    division: str = Field(..., description="A division interviewers belong to. Required parameter.")
-    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on. Required parameter.")
-    start_date_time: str = Field(..., description="Date and time of start of interview. Optional parameter.")
-    duration_hours: int = Field(..., description="Required duration of the availability in hours. Optional parameter.")
-    skills: list = Field(..., description="Skills that an interviewers must have to be able to conduct an interview. Optional parameter.")
+    team_size: int = Field(..., description="The number of interviewers in each team.")
+    division: str = Field(..., description="A division interviewers belong to.")
+    interview_level: str = Field(..., description="A level of interview. E.g. for Junior interview level we need interviewers level Junior or higher. For Middle - Middle and higher. And so on.")
+    start_date_time: str = Field(..., description="Date and time of start of interview.")
+    duration_hours: int = Field(..., description="Required duration of the availability in hours.")
+    skills: list = Field(..., description="Skills that an interviewers must have to be able to conduct an interview.")
 
 llm = ChatOpenAI(temperature=0.0, model_name="gpt-3.5-turbo", openai_api_key=OPENAI_API_KEY)
 
@@ -172,6 +208,11 @@ tools = [
         args_schema=GetInterviewerTeamsBySkills,
         description="A function to get all possible teams of interviewers by their skills."
     ),
+    StructuredTool.from_function(
+        func=get_interviewer_teams_in_time_slot,
+        args_schema=GetInterviewerTeamsInTimeSlot,
+        description="A function to get all possible teams of interviewers in requested time slot."
+    ),
     StructuredTool.from_function(
         func=get_interviewer_teams_by_skills_in_time_slot,
         args_schema=GetInterviewerTeamsBySkillsInTimeSlot,
@@ -211,9 +252,11 @@ def predict(message, history):
 
 examples = [
     "Hello! Tell me please about yourself: who are you, what is your purpose, what tools do you have?",
-    "List all employees",
     "I need a team of 2 interviewers from Java division who can interview Senior level newcomers",
-    "I need a team of 2 interviewers from Java division with Java and SQL skills who can interview Senior level newcomers in given time slot: start time is 2024-02-12T11:00:00, duration 1 hour"
+    "I need a team of 2 interviewers from Java division with Java and SQL skills who can interview Senior level newcomers",
+    "I need a team of 2 interviewers from Java division who can interview Senior level newcomers in given time slot: start time is 2024-02-12T11:00:00, duration 1 hour",
+    "I need a team of 2 interviewers from Java division with Java and SQL skills who can interview Senior level newcomers in given time slot: start time is 2024-02-12T11:00:00, duration 1 hour",
+    "List all interviewers"
 ]
 
 description = '''