Merge pull request #16 from dreadnode/ads/eng-1353-attributeerror-in-rigging_examplepy-regarding

fix: Ads/eng 1353 attributeerror in rigging examplepy regarding

Author: GangGreenTemperTatum

.gitignore

@@ -1,15 +1,110 @@
-/target
+### Rust ###
+/target/
+**/*.rs.bk
+*.pdb
+Cargo.lock
+# Keep Cargo.lock for binary projects, uncomment this line if it's a library
+
+### IDE and Editors ###
+# VSCode
+.vscode/*
+!.vscode/settings.json
+!.vscode/tasks.json
+!.vscode/launch.json
+!.vscode/extensions.json
+*.code-workspace
+
+# JetBrains IDEs (IntelliJ, CLion, etc.)
+.idea/
+*.iml
+*.iws
+*.ipr
+out/
+.idea_modules/
+
+# SublimeText
+*.sublime-workspace
+*.sublime-project
+
+# Mac
 .DS_Store
-.mypy_cache
-__pycache__
+.AppleDouble
+.LSOverride
+._*
+
+# Windows
+Thumbs.db
+ehthumbs.db
+Desktop.ini
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+.python-version
 
-/.venv
+# Jupyter Notebook
+.ipynb_checkpoints
+notebooks/
 
+# pytest
+.pytest_cache/
+.coverage
+htmlcov/
 
+### Project specific ###
 # proxy stuff
 burp.cer
 burp.pem
 install_burp.py
 
-# Local working files
-hans_test.py
+# Logs
+logs/
+*.log
+npm-debug.log*
+
+# Dependencies
+node_modules/
+
+# Local config
+.env.local
+.env.development.local
+.env.test.local
+.env.production.local
+
+# Build artifacts
+/dist
+/build
+
+# proxy stuff
+burp.cer
+burp.pem
+install_burp.py

examples/rigging_example.py

@@ -1,85 +1,59 @@
 import asyncio
-import requests
+import os
+from loguru import logger
 import rigging as rg
+from rigging import logging
 from rich import print
 
+os.environ["LOGFIRE_IGNORE_NO_CONFIG"] = "1"
+logging.configure_logging("DEBUG", None, "DEBUG")
 
-# we need to wrap the tools in a class that Rigging can understand
-class Wrapper(rg.Tool):
-    # we'll set these in the constructor
-    name = "_"
-    description = "_"
-
-    def __init__(self, tool: dict):
-        self.tool = tool
-        self.name = tool["name"]
-        self.description = tool["description"]
-
-        # declare dynamically the functions by their name
-        for function in tool["functions"]:
-            setattr(
-                Wrapper,
-                function["name"],
-                lambda self, *args, **kwargs: self._execute_function(
-                    function["name"], *args, **kwargs
-                ),
+
+async def run():
+    """Main function that runs the chat with RoboPages tools"""
+
+    try:
+        logger.info("Fetching tools from RoboPages server")
+        # Use the built-in robopages integration
+        tools = rg.integrations.robopages("http://localhost:8000")
+
+        logger.info(f"Fetched {len(tools)} tools from RoboPages server")
+
+        prompt = """
+        I need you to find all open ports on the local machine (127.0.0.1).
+        Please use the available tools to scan the ports and provide a summary of the results.
+
+        Be thorough but concise in your analysis. Present the information in a clear format.
+
+        After scanning, list all the open ports you found and what services might be running on them.
+        """
+
+        logger.info("Starting chat with model")
+        generator = rg.get_generator("gpt-4o")
+
+        chat = await generator.chat(prompt).using(*tools, force=True).run()
+
+        logger.info("Chat completed. Full conversation:")
+        for i, message in enumerate(chat.messages):
+            logger.info(f"Message {i + 1} ({message.role}):")
+            logger.info(
+                message.content[:200] + ("..." if len(message.content) > 200 else "")
             )
 
-    def _execute_function(self, func_name: str, *args, **kwargs):
-        print(f"executing {self.name}.{func_name}{kwargs} ...")
-        # execute the call via robopages and return the result to Rigging
-        return requests.post(
-            "http://localhost:8000/process",
-            json=[
-                {
-                    "type": "function",
-                    "function": {
-                        "name": func_name,
-                        "arguments": kwargs,
-                    },
-                }
-            ],
-        ).json()[0]["content"]
-
-    def get_description(self) -> rg.tool.ToolDescription:
-        """Creates a full description of the tool for use in prompting"""
-
-        return rg.tool.ToolDescription(
-            name=self.name,
-            description=self.description,
-            functions=[
-                rg.tool.ToolFunction(
-                    name=function["name"],
-                    description=function["description"],
-                    parameters=[
-                        rg.tool.ToolParameter(
-                            name=param["name"],
-                            type=param["type"],
-                            description=param["description"],
-                        )
-                        for param in function["parameters"]
-                    ],
-                )
-                for function in self.tool["functions"]
-            ],
-        )
-
-
-async def run(model: str):
-    # get the tools from the Robopages server and wrap each function for Rigging
-    tools = [
-        Wrapper(tool)
-        for tool in requests.get("http://localhost:8000/?flavor=rigging").json()
-    ]
-
-    chat = (
-        await rg.get_generator(model)
-        .chat("Find open ports on 127.0.0.1")
-        .using(*tools, force=True)
-        .run()
-    )
-
-    print(chat.last.content)
-
-
-asyncio.run(run("gpt-4o"))
+        print("\n--- RESULT ---\n")
+        print(chat.last.content)
+
+        return chat
+
+    except Exception as e:
+        logger.error(f"Error: {e}")
+        import traceback
+
+        traceback.print_exc()
+        return None
+
+
+if __name__ == "__main__":
+    chat = asyncio.run(run())
+    if chat:
+        print(chat.conversation)