Parse the messages to Anthropic format.

Parse LLM messages to OpenAI format.

Parse the messages to OpenAI format.

Ollama parse messages should be compatible with openai handling

Search for a function in the database. Args: identifier: the identifier of the function to search for Returns:

Remove any changes to the tests directory from the provided patch. This is to ensure that the model_patch does not disturb the repo's

Set the agent context for task execution in interactive mode.

Set the CLI console for this agent.

Starts/attaches to the container, mounts the workspace, copies tools, and starts the shell.

Validate the path for the str_replace_editor tool.

Add value at specified JSONPath.

Check existence of required Docker images for all instances. Updates self.image_status dict.

Release tool resources, mainly about BashTool object.

Copies the local tools directory to a fixed path inside the container.

Create lakeview display for a step.

Builds and executes a command inside the Docker container, with path translation.

Executes a command within the existing persistent shell.

Execute a task using the agent.

Implement the insert command, which inserts new_str at the specified line in the file content.

Implement the insert command, which inserts new_str at the specified line in the file content.

Insert class entry into db. Args: entry: the entry to insert Returns: None

Insert function entry including functions and class methodsinto db. Args: entry: the entry to insert Returns:

Check if the task is completed based on the response. Override for custom logic.

Helper function to load JSONL dataset using context manager.

Print the final execution summary.

Print lakeview summary of all completed steps.

Print a step update as it progresses.

Recursively visit the C AST and insert the entries into the database.

Recursively visit the C++ AST and insert the entries into the database.

Recursively visit the Java AST and insert the entries into the database.

Recursively visit the JavaScript AST and insert the entries into the database.

Recursively visit the Python AST and insert the entries into the database.

Remove value at specified JSONPath.

Run the interactive loop for rich console.

Run the interactive loop for simple console.

Search for a class in the ckg database.

Search for a class method in the ckg database.

Search for a function in the ckg database.

Set value at specified JSONPath.

Robust path translation function: Translate the host path into the corresponding path within the container.

Validate the input arguments and return a ThoughtData object.

View JSON file content or specific paths.

Helper function to write multi-issue problem statement using context manager.

A rewritten version of ollama chan

Clean up all MCP clients to prevent async context leaks.

Clean up MCP clients. Override in subclasses that use MCP.

Iterate over all the files in the CKG storage directory and delete the ones that are older than 1 week.

Trae Agent - LLM-based agent for software engineering tasks.

Ensure all tool resources are properly released.

Create the OpenAI client instance.

Execute the task and finalize trajectory recording.

Collect all generated patches and write predictions.json to results directory. Args: instance_ids: List of instance IDs

Get any extra headers needed for the API call.

Get the hash of the folder snapshot, to make sure that the CKG is up to date.

Get hash for git repository (clean or dirty).

Get the provider name for trajectory recording.

Get the service name for retry logging.

Get the system prompt for TraeAgent.

Get task input from user (for interactive mode). Returns: Task string or None if user wants to exit