Method input_variables_identifying_agent

integuru/agent.py:76–144

Identify input variables present in the cURL command

(self, state: AgentState)

        return state

    def input_variables_identifying_agent(self, state: AgentState) -> AgentState:
        """
        Identify input variables present in the cURL command
        """
        in_process_node_id = state[self.IN_PROCESS_NODE_KEY]
        curl = self.dag_manager.graph.nodes[in_process_node_id]["content"]["key"].to_curl_command()
        input_variables = state[self.INPUT_VARIABLES_KEY]
        if not input_variables:
            return state
        
        function_def = {
            "name": "identify_input_variables",
            "description": "Identify input variables present in the cURL command.",
            "parameters": {
                "type": "object",
                "properties": {
                    "identified_variables": {
                        "type": "array",
                        "items": {
                            "type": "object",
                            "properties": {
                                "variable_name": {"type": "string", "description": "The original key of the variable"},
                                "variable_value": {"type": "string", "description": "The exact version of the variable that is present in the cURL command. This should closely match the value in the provided Input Variables."}
                            },
                            "required": ["variable_name", "variable_value"]
                        },
                        "description": "A list of identified variables and their values."
                    }
                },
                "required": ["identified_variables"]
            }
        }


        prompt = f"""
        cURL: {curl}
        Input Variables: {input_variables}

        Task:
        Identify which input variables (the value in the key-value pair) from the Input Variables provided above are present in the cURL command.

        Important:
        - If an input variable is found in the cURL, include it in the output.
        - Do not include variables that are not provided above.
        - The key of the input variable is a description of the variable.
        - The value is the value that should closely match the value in the cURL command. No substitutions.

        """


        response = llm.get_instance().invoke(
            prompt,
            functions=[function_def],
            function_call={"name": "identify_input_variables"}
        )

        function_call = response.additional_kwargs.get('function_call', {})
        arguments = json.loads(function_call.get('arguments', '{}'))