Replace stub with call of genetic algorythm at AWS endpoint

app.py

@@ -9,6 +9,7 @@ from langchain.agents import AgentExecutor, create_openai_functions_agent
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.messages import AIMessage, HumanMessage
 from langchain.tools import StructuredTool
+from datetime import datetime, timedelta
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
@@ -35,53 +36,144 @@ def get_all_employees() -> str:
     """
     return repo_get_all_employees_from_database()
 
-def get_employees(number_of_employees: int, start_time: str, duration_hours: int) -> str:
+def get_interviewer_teams(
+        team_size: int,
+        division: str,
+        interview_level: str) -> str:
     """
-    A function to get a required number_of_employees that are available from tart_time during specified duration_hours.
+    A function to get all possible teams of interviewers.
     Args:
-        number_of_employees (int): Required number of employees.
-        start_time (str): Required start time.
-        duration_hours (int): Required duration of the availability in hours. 
+        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.
+
     Returns: 
-        str: Employees list in json.
-    """
-    method_return_value_stub = '''
-    {
-      "employees": [
-        {
-          "id": 100,
-          "name": "Lana Kane"
-        }
-      ]
-    }
-    '''
-    return method_return_value_stub
+        str: A list of interviewers teams in json.
+    """
+    return get_interviewer_teams_by_skills(
+        team_size=team_size, 
+        division=division, 
+        interview_level=interview_level,
+        skills=[])
+
+def get_interviewer_teams_by_skills(
+        team_size: int,
+        division: str,
+        interview_level: str,
+        skills: list) -> str:
+    """
+    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.
 
-class GetAllEmployees(BaseModel):
+    Returns: 
+        str: A list of interviewers teams in json.
     """
-    Pydantic arguments schema for get_all_employees function
+    return get_interviewer_teams_by_skills_in_time_slot(
+        team_size=team_size, 
+        division=division, 
+        interview_level=interview_level, 
+        start_date_time=None,
+        duration_hours=0, 
+        skills=skills)
+
+def get_interviewer_teams_by_skills_in_time_slot(
+        team_size: int,
+        division: str,
+        interview_level: str,
+        start_date_time: str, 
+        duration_hours: int, 
+        skills: list) -> str:
     """
+    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.
 
-class GetEmployees(BaseModel):
+    Returns: 
+        str: A list of interviewers teams in json.
+    """
+    
+    params = "?teamSize={team_size}&division={division}&interviewLevel={interview_level}".format(
+        team_size=team_size, 
+        division=division, 
+        interview_level=interview_level)
+
+    if start_date_time is not None:
+        end_date_time = start_date_time
+        start_date_time_obj = datetime.fromisoformat(start_date_time)
+        end_date_time_obj = start_date_time_obj + timedelta(hours=duration_hours)
+        end_date_time = end_date_time_obj.isoformat()
+        optional_params = "&startDateTime={start}&endDateTime={end}".format(start=start_date_time, end=end_date_time)
+        params += optional_params
+    
+    basic_url = "http://18.133.247.78/search/interviewerTeams"    
+    url = basic_url + params
+    print(params)
+    return requests.get(url).content
+
+class GetAllInterviewers(BaseModel):
     """
-    Pydantic arguments schema for get_employees function
+    Pydantic arguments schema for get_all_interviewers function
     """
-    number_of_employees: int = Field(..., description="Required number of employees")
-    start_time: str = Field(..., description="Required start time")
-    duration_hours: int = Field(..., description="Required duration of the availability in hours")
 
-llm = ChatOpenAI(temperature=1.0, model_name="gpt-3.5-turbo", openai_api_key=OPENAI_API_KEY)
+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.")  
+
+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.")
+
+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.")
+
+llm = ChatOpenAI(temperature=0.0, model_name="gpt-3.5-turbo", openai_api_key=OPENAI_API_KEY)
 
 tools = [
     StructuredTool.from_function(
-        func=get_all_employees,
-        args_schema=GetAllEmployees,
-        description="A function to get a list of all employees from database."
+        func=get_all_interviewers,
+        args_schema=GetAllInterviewers,
+        description="A function to get a list of all interviewers from database."
+    ),
+    StructuredTool.from_function(
+        func=get_interviewer_teams,
+        args_schema=GetInterviewerTeams,
+        description="A function to get all possible teams of interviewers."
+    ),
+    StructuredTool.from_function(
+        func=get_interviewer_teams_by_skills,
+        args_schema=GetInterviewerTeamsBySkills,
+        description="A function to get all possible teams of interviewers by their skills."
     ),
     StructuredTool.from_function(
-        func=get_employees,
-        args_schema=GetEmployees,
-        description="A function to get a required number_of_employees that are available from tart_time during specified duration_hours."
+        func=get_interviewer_teams_by_skills_in_time_slot,
+        args_schema=GetInterviewerTeamsBySkillsInTimeSlot,
+        description="A function to get all possible teams of interviewers by their skills in requested time slot."
     )
 ]
 
@@ -116,9 +208,10 @@ def predict(message, history):
     return gpt_output
 
 examples = [
-    "Tell me please about yourself: who are you, what is your purpose, what tools do you have?",
+    "Hello! Tell me please about yourself: who are you, what is your purpose, what tools do you have?",
     "List all employees",
-    "I need 1 employee in given time slot: start time is March 11 2024 2 pm, duration 1 hour"
+    "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"
 ]
 
 description = '''