Skip to content

allow running hackingbuddygpt with llama-style models again #30

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 6 commits into from
Apr 24, 2024
Merged

allow running hackingbuddygpt with llama-style models again #30

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
88324d4
capture stderr/stdout over SSH
andreashappe Apr 23, 2024
eef91c7
update readme.md
andreashappe Apr 23, 2024
54c044b
the 'minimal' usecase is actually not an abstrct base class
andreashappe Apr 24, 2024
a9bf484
mark the `perform_round` method as abstract
andreashappe Apr 24, 2024
66e8e10
reintroduce quick hack so that we can test non-openai LLMs easily
andreashappe Apr 24, 2024
b45bc77
Merge branch 'main' of github.com:ipa-lab/hackingBuddyGPT into other_…
andreashappe Apr 24, 2024
File filter

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
capture stderr/stdout over SSH 
I need to find out why this solution works while others don't. And
make it more readable..
  • Loading branch information
@andreashappe
andreashappe committed Apr 23, 2024
commit 88324d45f1148a1898a438f619964a19ca429883
10 changes: 7 additions & 3 deletions capabilities/ssh_run_command.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

from invoke import Responder

from io import StringIO
from utils import SSHConnection
from .capability import Capability

Expand All @@ -28,15 +29,18 @@ def __call__(self, command: str) -> Tuple[str, bool]:
response=self.conn.password + '\n',
)

out = StringIO()

try:
stdout, stderr, rc = self.conn.run(command, pty=True, warn=True, watchers=[sudo_pass], timeout=10)
resp = self.conn.run(command, pty=True, warn=True, out_stream=out, watchers=[sudo_pass], timeout=10)
except Exception as e:
print("TIMEOUT! Could we have become root?")
stdout, stderr, rc = "", "", -1
out.seek(0)
tmp = ""
last_line = ""
for line in stdout.splitlines():
for line in out.readlines():
if not line.startswith('[sudo] password for ' + self.conn.username + ':'):
line.replace("\r", "")
last_line = line
tmp = tmp + line

Expand Down