Merge pull request #47 from dreadnode/example/add-robopages-tool-server-example

feat: add robopages native tool server example

Author: GangGreenTemperTatum

README.md

@@ -140,7 +140,8 @@ Want more?
 - OverTheWire Bandit Agent: [**bandit.py**](examples/bandit.py)
 - Damn Vulnerable Restaurant Agent: [**dvra.py**](examples/dvra.py)
 - RAG Pipeline: [**rag.py**](examples/rag.py) (from [kyleavery](https://github.com/kyleavery/))
- 
+- Integrating dreadnode-owned [robopages](https://github.com/dreadnode/robopages-cli) as a tool server (basic nmap scan example): [**rigging_example.py**](https://github.com/dreadnode/robopages-cli/blob/main/examples/rigging_example.py)
+
 ## Documentation
 
 **[docs.dreadnode.io](https://docs.dreadnode.io/open-source/rigging/intro)** has everything you need.

examples/robopages.py

@@ -0,0 +1,27 @@
+import asyncio
+import os
+
+import litellm
+import logfire
+
+import rigging as rg
+
+# Constants
+SYSTEM_PROMPT = "Enumerate all open TCP ports on 127.0.0.1 and provide a vulnerability report"
+
+# LOGFIRE_PROJECT = "rigging-demos"
+logfire.configure()
+os.environ.setdefault("LOGFIRE_TOKEN", "")  # (1)!
+litellm.callbacks = ["logfire"]
+
+rg.logging.configure_logging("debug")
+
+
+async def main():
+    tools = rg.integrations.robopages("http://localhost:8000")
+    chat = await rg.get_generator("gpt-4").chat(SYSTEM_PROMPT).using(*tools).run()
+    print(chat.conversation)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())