Skip to content

Commit 72de4a0

Browse files
Diana StraussDiana Strauss
authored andcommitted
Added changes suggested by Code review
1 parent 84c2e7e commit 72de4a0

File tree

6 files changed

+3
-4
lines changed

6 files changed

+3
-4
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ Feel free to connect or talk with us on various platforms:
3838
- Andreas Happe: [github](https://github.com/andreashappe), [linkedin](https://at.linkedin.com/in/andreashappe), [twitter/x](https://twitter.com/andreashappe), [Google Scholar](https://scholar.google.at/citations?user=Xy_UZUUAAAAJ&hl=de)
3939
- Juergen Cito, [github](https://github.com/citostyle), [linkedin](https://at.linkedin.com/in/jcito), [twitter/x](https://twitter.com/citostyle), [Google Scholar](https://scholar.google.ch/citations?user=fj5MiWsAAAAJ&hl=en)
4040
- Manuel Reinsperger, [github](https://github.com/Neverbolt), [linkedin](https://www.linkedin.com/in/manuel-reinsperger-7110b8113/), [twitter/x](https://twitter.com/neverbolt)
41+
- Diana Strauss , [github](https://github.com/DianaStrauss), [linkedin](https://www.linkedin.com/in/diana-s-a853ba20a/)
4142
- we have a [discord server were we talk about all things AI + Offensive Security](https://discord.gg/vr4PhSM8yN)
4243

4344
## Existing Agents/Usecases

docs/web-api-pentest.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ The goal of this use-case is to explore REST API security. It is currently very
1414
This is a simple example run of the `simple_web_api_documenation` using GPT-3.5-turbo to test the REST API https://jsonplaceholder.typicode.com.
1515
![web_api_documentation.png](..%2F..%2F..%2F..%2FPictures%2Fweb_api_documentation.png)
1616

17-
OpenAPI specifications that have been created with this can be found under [openapi_spec](../openapi_spec).
17+
OpenAPI specifications that have been created with this can be found under [openapi_spec](openapi_spec).
1818

1919
This is a simple example run of the `simple_web_api_testing` using GPT-3.5-turbo to test the REST API https://jsonplaceholder.typicode.com.
2020
![web_api_testing.png](..%2F..%2F..%2F..%2FPictures%2Fweb_api_testing.png)

usecases/web/simple.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
@dataclass
2727
class MinimalWebTesting(RoundBasedUseCase):
2828
llm: OpenAILib
29-
host: str = parameter(desc="The host to test", default="https://api.restful-api.dev/objects")
29+
host: str = parameter(desc="The host to test", default="http://localhost")
3030
flag_format_description: str = parameter(desc="Description of the flag provided to the LLM", default="a string starting with 'FLAG.' and ending with '.GALF'")
3131
flag_template: str = parameter(desc="The template of the flags, whereby {flag} is replaced with the flags", default="FLAG.{flag}.GALF")
3232
flags: str = parameter(desc="A comma (,) separated list of flags to find", default="hostname,dir,username,rootfile,secretfile,adminpass")
@@ -77,8 +77,6 @@ def perform_round(self, turn: int):
7777
result = response.execute()
7878
self.console.print(Panel(result, title="tool"))
7979
self._prompt_history.append(tool_message(result, tool_call_id))
80-
for i in self._prompt_history:
81-
print(f'i:{i}')
8280

8381
self.log_db.add_log_query(self._run_id, turn, command, result, answer)
8482
return self._all_flags_found

0 commit comments

Comments
 (0)