Method end_url_identify_agent

integuru/agent.py:38–74

Identify the URL responsible for a specific action

(self, state: AgentState)

        self.dag_manager: DAGManager = DAGManager()

    def end_url_identify_agent(self, state: AgentState) -> AgentState:
        """
        Identify the URL responsible for a specific action
        """
        function_def = {
            "name": "identify_end_url",
            "description": "Identify the URL responsible for a specific action",
            "parameters": {
                "type": "object",
                "properties": {
                    "url": {
                        "type": "string",
                        "description": f"The URL responsible for {self.prompt}"
                    }
                },
                "required": ["url"]
            }
        }

        prompt = f"""
        {self.har_urls}
        Task:
        Given the above list of URLs, request types, and response formats, find the URL responsible for the action below:
        {self.prompt}
        """

        response = llm.get_instance().invoke(
            prompt,
            functions=[function_def],
            function_call={"name": "identify_end_url"}
        )
        
        function_call = response.additional_kwargs['function_call']
        end_url = json.loads(function_call['arguments'])['url']

        state[self.ACTION_URL_KEY] = end_url
        return state

    def input_variables_identifying_agent(self, state: AgentState) -> AgentState:
        """