Function rawdog

src/rawdog/__main__.py:11–53

(prompt: str, config, llm_client)

def rawdog(prompt: str, config, llm_client):
    verbose = config.get("dry_run")
    retries = int(config.get("retries"))
    _continue = True
    _first = True
    while _continue is True:
        error, script, output = "", "", ""
        try:
            if _first:
                message, script = llm_client.get_script(prompt, stream=verbose)
                _first = False
            else:
                message, script = llm_client.get_script(stream=verbose)
            if script:
                if verbose:
                    print(f"\n{80 * '-'}")
                    if (
                        input("Execute script in markdown block? (Y/n): ")
                        .strip()
                        .lower()
                        == "n"
                    ):
                        llm_client.add_message("user", "User chose not to run script")
                        break
                output, error = execute_script(script, llm_client)
            elif message:
                print(message)
        except KeyboardInterrupt:
            break

        _continue = (
            output and output.strip().endswith("CONTINUE") or error and retries > 0
        )
        if error:
            retries -= 1
            llm_client.add_message("user", f"Error: {error}")
            print(f"Error: {error}")
            if script and not verbose:
                print(f"{80 * '-'}\n{script}\n{80 * '-'}")
        if output:
            llm_client.add_message("user", f"LAST SCRIPT OUTPUT:\n{output}")
            if verbose or not _continue:
                print(output)


def banner():