Merging PR_218 openai_rev package with new streamlit chat app

venv/lib/python3.9/site-packages/streamlit/code_util.py

@@ -0,0 +1,87 @@
+# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022)
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""A bunch of useful code utilities."""
+
+import re
+
+
+def extract_args(line):
+    """Parse argument strings from all outer parentheses in a line of code.
+
+    Parameters
+    ----------
+    line : str
+        A line of code
+
+    Returns
+    -------
+    list of strings
+        Contents of the outer parentheses
+
+    Example
+    -------
+    >>> line = 'foo(bar, baz), "a", my(func)'
+    >>> extract_args(line)
+    ['bar, baz', 'func']
+
+    """
+    stack = 0
+    startIndex = None
+    results = []
+
+    for i, c in enumerate(line):
+        if c == "(":
+            if stack == 0:
+                startIndex = i + 1
+            stack += 1
+        elif c == ")":
+            stack -= 1
+            if stack == 0:
+                results.append(line[startIndex:i])
+    return results
+
+
+def get_method_args_from_code(args, line):
+    """Parse arguments from a stringified arguments list inside parentheses
+
+    Parameters
+    ----------
+    args : list
+        A list where it's size matches the expected number of parsed arguments
+    line : str
+        Stringified line of code with method arguments inside parentheses
+
+    Returns
+    -------
+    list of strings
+        Parsed arguments
+
+    Example
+    -------
+    >>> line = 'foo(bar, baz, my(func, tion))'
+    >>>
+    >>> get_method_args_from_code(range(0, 3), line)
+    ['bar', 'baz', 'my(func, tion)']
+
+    """
+    line_args = extract_args(line)[0]
+
+    # Split arguments, https://stackoverflow.com/a/26634150
+    if len(args) > 1:
+        inputs = re.split(r",\s*(?![^(){}[\]]*\))", line_args)
+        assert len(inputs) == len(args), "Could not split arguments"
+    else:
+        inputs = [line_args]
+    return inputs