Skip to content

Commit

Permalink
Potential syntax error fix
Browse files Browse the repository at this point in the history
  • Loading branch information
KillianLucas committed Dec 3, 2024
1 parent 848252c commit eae8cf2
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions scripts/wtf.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,8 @@ def stop(self):
if os.environ.get("SHELL"):
shell = "Shell: " + os.environ.get("SHELL")

CUSTOM_MESSAGE_SYSTEM_MESSAGE = f"""
CUSTOM_MESSAGE_SYSTEM_MESSAGE = (
r"""
You are a fast, efficient AI assistant for terminal and coding tasks. When summoned, you will:
Expand All @@ -132,15 +133,18 @@ def stop(self):
- If more info is needed, provide a command to gather it (e.g., grep).
- Focus on the user's FINAL query and ADDRESS NOTHING ELSE, using terminal history for context if relevant.
- For multi-step solutions, explain briefly and provide the first or combined command.
- Prioritize addressing the user's specific request (at the END, after "wtf") efficiently.
- Prioritize addressing the user's specific request (at the END, after "wtf") efficiently."""
+ f"""
User's System: {platform.system()}
CWD: {os.getcwd()}
{shell}
""".strip()
)

LOCAL_SYSTEM_MESSAGE = f"""
LOCAL_SYSTEM_MESSAGE = (
r"""
You're a fast AI assistant for terminal issues. You must:
1. Scan terminal history
Expand All @@ -161,14 +165,16 @@ def stop(self):
We need to fix the file permissions on config.yml.
```bash
chmod 644 config.yml
```
```"""
+ f"""
User's System: {platform.system()}
CWD: {os.getcwd()}
{"Shell: " + os.environ.get('SHELL') if os.environ.get('SHELL') else ''}
Now, it's your turn:
"""
)


def main():
Expand Down

0 comments on commit eae8cf2

Please sign in to comment.