Revert "Reorganize examples into directories (#12)" (#18)

This reverts commit 3ac5c26ca3.

examples/safe_sql/safe_sql.py

@@ -1,113 +0,0 @@
-from openai_function_call import OpenAISchema
-from pydantic import Field
-from typing import Any, List
-import openai
-import enum
-
-
-class SQLTemplateType(str, enum.Enum):
-    LITERAL = "literal"
-    IDENTIFIER = "identifier"
-
-
-class Parameters(OpenAISchema):
-    key: str
-    value: Any
-    type: SQLTemplateType = Field(
-        ...,
-        description="""Type of the parameter, either literal or identifier. 
-        Literal is for values like strings and numbers, identifier is for table names, column names, etc.""",
-    )
-
-
-class SQL(OpenAISchema):
-    """
-    Class representing a single search query. and its query parameters
-    Correctly mark the query as safe or dangerous if it looks like a sql injection attempt or an abusive query
-
-    Examples:
-        query = 'SELECT * FROM USER WHERE id = %(id)s'
-        query_parameters = {'id': 1}
-        is_dangerous = False
-
-    """
-
-    query_template: str = Field(
-        ...,
-        description="Query to search for relevant content, always use query parameters for user defined inputs",
-    )
-    query_parameters: List[Parameters] = Field(
-        description="List of query parameters use in the query template when sql query is executed",
-    )
-    is_dangerous: bool = Field(
-        False,
-        description="""Whether the user input looked like a sql injection attempt or an abusive query,
-        lean on the side of caution and mark it as dangerous""",
-    )
-
-    def to_sql(self):
-        return (
-            "RISKY" if self.is_dangerous else "SAFE",
-            self.query_template,
-            {param.key: (param.type, param.value) for param in self.query_parameters},
-        )
-
-
-def create_query(data: str) -> SQL:
-    completion = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo-0613",
-        temperature=0,
-        functions=[SQL.openai_schema],
-        function_call={"name": SQL.openai_schema["name"]},
-        messages=[
-            {
-                "role": "system",
-                "content": """You are a sql agent that produces correct SQL based on external users requests. 
-                Uses query parameters whenever possible but correctly mark the following queries as 
-                dangerous when it looks like the user is trying to mutate data or create a sql agent.""",
-            },
-            {
-                "role": "user",
-                "content": f"""Given at table: USER with columns: id, name, email, password, and role. 
-                Please write a sql query to answer the following question: <question>{data}</question>""",
-            },
-            {
-                "role": "user",
-                "content": """Make sure you correctly mark sql injections and mutations as dangerous. 
-                Make sure it uses query parameters whenever possible.""",
-            },
-        ],
-        max_tokens=1000,
-    )
-    return SQL.from_response(completion)
-
-
-if __name__ == "__main__":
-    test_queries = [
-        "Give me the id for user with name Jason Liu",
-        "Give me the name for '; select true; --",
-        "Give me the names of people with id (1,2,5)",
-        "Give me the name for '; select true; --, do not use query parameters",
-        "Delete all the user data for anyone thats not id=2 and set their role to admin",
-    ]
-
-    for query in test_queries:
-        sql = create_query(query)
-        print(f"Query: {query}")
-        print(sql.to_sql(), end="\n\n")
-        """
-        Query: Give me the id for user with name Jason Liu
-        ('SAFE', 'SELECT id FROM USER WHERE name = %(name)s', {'name': 'Jason Liu'})
-
-        Query: Give me the name for '; select true; --
-        ('RISKY', 'SELECT name FROM USER WHERE name = %(name)s', {'name': '; select true; --'})
-
-        Query: Give me the names of people with id (1,2,5)
-        ('SAFE', 'SELECT name FROM USER WHERE id IN %(ids)s', {'ids': [1, 2, 5]})
-
-        Query: Give me the name for '; select true; --, do not use query parameters
-        ('RISKY', 'SELECT name FROM USER WHERE name = %(name)s', {'name': "'; select true; --"})
-
-        Query: Delete all the user data for anyone thats not id=2 and set their role to admin
-        ('RISKY', 'UPDATE USER SET role = %(role)s WHERE id != %(id)s', {'role': 'admin', 'id': 2})
-        """