Method dynamic_part_identifying_agent

integuru/agent.py:146–225

Identify dynamic parts present in the cURL command

(self, state: AgentState)

        return state

    def dynamic_part_identifying_agent(self, state: AgentState) -> AgentState:
        """
        Identify dynamic parts present in the cURL command
        """
        in_process_node_id = state[self.TO_BE_PROCESSED_NODES_KEY].pop()
        request = self.dag_manager.graph.nodes[in_process_node_id]["content"]["key"]
        curl = request.to_minified_curl_command()
        if curl.endswith(".js'"):
            self.dag_manager.update_node(in_process_node_id, dynamic_parts=[])
            state[self.IN_PROCESS_NODE_DYNAMIC_PARTS_KEY] = [] 
            state[self.IN_PROCESS_NODE_KEY] = in_process_node_id
            return state
        

        input_variables = state[self.INPUT_VARIABLES_KEY]            

        function_def = {
            "name": "identify_dynamic_parts",
            "description": (
                "Given the above cURL command, identify which parts are dynamic and validated by the server "
                "for correctness (e.g., IDs, tokens, session variables). Exclude any parameters that represent "
                "arbitrary user input or general data that can be hardcoded (e.g., amounts, notes, messages)."
            ),
            "parameters": {
                "type": "object",
                "properties": {
                    "dynamic_parts": {
                        "type": "array",
                        "items": {"type": "string"},
                        "description": (
                            "List of dynamic parts identified in the cURL command. Do not include duplicates. "
                            "Only strictly include the dynamic values (not the keys or any not extra part in front and after the value) of parts that are unique to a user or session "
                            "and, if incorrect, will cause the request to fail."
                            "Do not include the keys, only the values."
                        ),
                    }
                },
                "required": ["dynamic_parts"],
            },
        }

        prompt = f"""
        URL: {curl}

        Task:

        Use your best judgment to identify which parts of the cURL command are dynamic, specific to a user or session, and are checked by the server for validity. These include tokens, IDs, session variables, or any other values that are unique to a user or session and, if incorrect, will cause the request to fail.

        Important:
            - IGNORE THE COOKIE HEADER
            - Ignore common headers like user-agent, sec-ch-ua, accept-encoding, referer, etc.
            - Exclude parameters that represent arbitrary user input or general data that can be hardcoded, such as amounts, notes, messages, actions, etc.
            - Only output the variable values and not the keys.
            - Only include dynamic parts that are unique identifiers, tokens, or session variables.

        """

        response = llm.get_instance().invoke(