fluzzi32/connpy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bug fixes

Browse Source
This commit is contained in:
fluzzi 2023-05-23 16:15:03 -03:00
parent 51f86f214a
commit acbfb03b10
3 changed files with 53 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
connpy/_version.py
View File

@ -1,2 +1,2 @@
__version__ = "3.2.6"
__version__ = "3.2.7"

18
connpy/ai.py
View File

@ -139,6 +139,16 @@ Users will typically use words like verify, check, make sure, or similar to refe
self.model = model
self.temp = temp
def process_string(self, s):
if s.startswith('[') and s.endswith(']') and not (s.startswith("['") and s.endswith("']")) and not (s.startswith('["') and s.endswith('"]')):
# Extract the content inside square brackets and split by comma
content = s[1:-1].split(',')
# Add single quotes around each item and join them back together with commas
new_content = ', '.join(f"'{item.strip()}'" for item in content)
# Replace the old content with the new content
s = '[' + new_content + ']'
return s
def _retry_function(self, function, max_retries, backoff_num, *args):
#Retry openai requests
retries = 0
@ -176,6 +186,7 @@ Users will typically use words like verify, check, make sure, or similar to refe
elif value.lower() == "none":
value = None
if key == "filter":
value = self.process_string(value)
value = ast.literal_eval(value)
#store in dictionary
info_dict[key] = value
@ -229,10 +240,10 @@ Users will typically use words like verify, check, make sure, or similar to refe
for key, value in nodes.items():
tags = value.get('tags', {})
try:
os_value = tags.get('os', '')
except:
os_value = ""
if os_value := tags.get('os'):
output_list.append(f"{key}: {os_value}")
except:
pass
output_str = "\n".join(output_list)
command_input = f"input: {user_input}\n\nDevices:\n{output_str}"
message = []
@ -411,6 +422,7 @@ Users will typically use words like verify, check, make sure, or similar to refe
output["args"] = {}
output["args"]["commands"] = commands["response"]["commands"]
output["args"]["vars"] = commands["response"]["variables"]
output["nodes"] = [item for item in output["nodes"] if output["args"]["vars"].get(item)]
if original["response"].get("expected"):
output["args"]["expected"] = original["response"]["expected"]
output["action"] = "test"

40
docs/connpy/index.html
View File

@ -706,6 +706,16 @@ Users will typically use words like verify, check, make sure, or similar to refe
self.model = model
self.temp = temp
def process_string(self, s):
if s.startswith('[') and s.endswith(']') and not (s.startswith("['") and s.endswith("']")) and not (s.startswith('["') and s.endswith('"]')):
# Extract the content inside square brackets and split by comma
content = s[1:-1].split(',')
# Add single quotes around each item and join them back together with commas
new_content = ', '.join(f"'{item.strip()}'" for item in content)
# Replace the old content with the new content
s = '[' + new_content + ']'
return s
def _retry_function(self, function, max_retries, backoff_num, *args):
#Retry openai requests
retries = 0
@ -743,6 +753,7 @@ Users will typically use words like verify, check, make sure, or similar to refe
elif value.lower() == "none":
value = None
if key == "filter":
value = self.process_string(value)
value = ast.literal_eval(value)
#store in dictionary
info_dict[key] = value
@ -796,10 +807,10 @@ Users will typically use words like verify, check, make sure, or similar to refe
for key, value in nodes.items():
tags = value.get('tags', {})
try:
os_value = tags.get('os', '')
except:
os_value = ""
if os_value := tags.get('os'):
output_list.append(f"{key}: {os_value}")
except:
pass
output_str = "\n".join(output_list)
command_input = f"input: {user_input}\n\nDevices:\n{output_str}"
message = []
@ -978,6 +989,7 @@ Users will typically use words like verify, check, make sure, or similar to refe
output["args"] = {}
output["args"]["commands"] = commands["response"]["commands"]
output["args"]["vars"] = commands["response"]["variables"]
output["nodes"] = [item for item in output["nodes"] if output["args"]["vars"].get(item)]
if original["response"].get("expected"):
output["args"]["expected"] = original["response"]["expected"]
output["action"] = "test"
@ -1142,6 +1154,7 @@ Users will typically use words like verify, check, make sure, or similar to refe
output["args"] = {}
output["args"]["commands"] = commands["response"]["commands"]
output["args"]["vars"] = commands["response"]["variables"]
output["nodes"] = [item for item in output["nodes"] if output["args"]["vars"].get(item)]
if original["response"].get("expected"):
output["args"]["expected"] = original["response"]["expected"]
output["action"] = "test"
@ -1213,6 +1226,26 @@ Users will typically use words like verify, check, make sure, or similar to refe
return output</code></pre>
</details>
</dd>
<dt id="connpy.ai.process_string"><code class="name flex">
<span>def <span class="ident">process_string</span></span>(<span>self, s)</span>
</code></dt>
<dd>
<div class="desc"></div>
<details class="source">
<summary>
<span>Expand source code</span>
</summary>
<pre><code class="python">def process_string(self, s):
if s.startswith(&#39;[&#39;) and s.endswith(&#39;]&#39;) and not (s.startswith(&#34;[&#39;&#34;) and s.endswith(&#34;&#39;]&#34;)) and not (s.startswith(&#39;[&#34;&#39;) and s.endswith(&#39;&#34;]&#39;)):
# Extract the content inside square brackets and split by comma
content = s[1:-1].split(&#39;,&#39;)
# Add single quotes around each item and join them back together with commas
new_content = &#39;, &#39;.join(f&#34;&#39;{item.strip()}&#39;&#34; for item in content)
# Replace the old content with the new content
s = &#39;[&#39; + new_content + &#39;]&#39;
return s</code></pre>
</details>
</dd>
</dl>
</dd>
<dt id="connpy.configfile"><code class="flex name class">
@ -4369,6 +4402,7 @@ tasks:
<ul class="">
<li><code><a title="connpy.ai.ask" href="#connpy.ai.ask">ask</a></code></li>
<li><code><a title="connpy.ai.confirm" href="#connpy.ai.confirm">confirm</a></code></li>
<li><code><a title="connpy.ai.process_string" href="#connpy.ai.process_string">process_string</a></code></li>
</ul>
</li>
<li>