Skip to content

Commit 848252c

Browse files
committed
Potential syntax error fix
1 parent 1510d85 commit 848252c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

scripts/wtf.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,10 @@ def stop(self):
112112
113113
"""
114114

115+
shell = ""
116+
if os.environ.get("SHELL"):
117+
shell = "Shell: " + os.environ.get("SHELL")
118+
115119
CUSTOM_MESSAGE_SYSTEM_MESSAGE = f"""
116120
117121
You are a fast, efficient AI assistant for terminal and coding tasks. When summoned, you will:
@@ -132,9 +136,9 @@ def stop(self):
132136
133137
User's System: {platform.system()}
134138
CWD: {os.getcwd()}
135-
{"Shell: " + os.environ.get('SHELL') if os.environ.get('SHELL') else ''}
139+
{shell}
136140
137-
"""
141+
""".strip()
138142

139143
LOCAL_SYSTEM_MESSAGE = f"""
140144
You're a fast AI assistant for terminal issues. You must:

0 commit comments

Comments
 (0)