Update README.md

README.md

@@ -76,21 +76,32 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 model = AutoModelForCausalLM.from_pretrained("PipableAI/pip-library-etl-1.3b").to(device)
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-library-etl-1.3b")
-prompt = f"""<example_response>
---code:def function_2(x): return x / 2
---question:Document the python code above giving function description ,parameters and return type and example how to call the function.
+prompt = f"""
+<example_response>
+--code:def divide_by_two(x: float) -> float: return x / 2
+--question:Document the python code above giving function description ,parameters and return type and example on how to call the function
 --doc:
-    Description:This function takes a number and divides it by 2.
-    Parameters:
-    - x (numeric): The input value to be divided by 2.
-    Returns:
-    - float: The result of x divided by 2
-    Example:
-    To call the function, use the following code:
-    function2(1.0)</example_response>
+Description: This function divides a given number by 2.
+Parameters:
+- x (float): The input value to be divided by 2.
+Returns:
+- float: The result of x divided by 2.
+Example:
+divide_by_two(1.0)
+</example_response>
 <function_code>
-def example_function(x):
-    return x * 2
+def download_file(shared_url, destination):
+    try:
+        if not shared_url.startswith("https://drive.google.com"):
+            raise ValueError("Please provde a valid google drive link.")
+        file_id = shared_url.split("/d/")[1]
+        file_id = file_id.split("/")[0]
+        url = f"https://drive.google.com/uc?id={file_id}"
+        gdown.download(url, destination, quiet=False)
+    except Exception as e:
+        print(f"Error downloading file from Google Drive as {e}")
+        raise e
+"""
 </function_code>
 <instructions> 
 1. In the examples while calling function use the name mentioned after `def ` in the above function_code.
@@ -121,18 +132,18 @@ print(doc)
 ### 1. Code Documentation
 ### prompt
 ```python
-prompt =''' <example_response>
---code:def function_2(x): return x / 2
---question:Document the python code above giving function description ,parameters and return type and example how to call the function.
+prompt ='''<example_response>
+--code:def divide_by_two(x: float) -> float: return x / 2
+--question:Document the python code above giving function description ,parameters and return type and example on how to call the function
 --doc:
-    Description:This function takes a number and divides it by 2.
-    Parameters:
-    - x (numeric): The input value to be divided by 2.
-    Returns:
-    - float: The result of x divided by 2
-    Example:
-    To call the function, use the following code:
-    function2(1.0)</example_response>
+Description: This function divides a given number by 2.
+Parameters:
+- x (float): The input value to be divided by 2.
+Returns:
+- float: The result of x divided by 2.
+Example:
+divide_by_two(1.0)
+</example_response>
 <function_code>def _plot_bounding_polygon(
     polygons_coordinates, output_html_path="bounding_polygon_map.html"
 ):
@@ -211,23 +222,32 @@ prompt =''' <example_response>
 ### prompt
 ```python
 prompt = """Generate a simple SQL query from the schema mentioned for the following question.
-<schema>CREATE TABLE department (Department_ID number,
-  Name text,
-  Creation text,
-  Ranking number,
-  Budget_in_Billions number,
-  Num_Employees number);
-
-  CREATE TABLE head (head_ID number,
-  name text,
-  born_state text,
-  age number);
-
-  CREATE TABLE management (department_ID number,
-  head_ID number,
-  temporary_acting text);</schema>
+<schema>
+CREATE TABLE department (
+  Department_ID number,         -- Unique identifier for the department
+  Name text,                     -- Name of the department
+  Creation text,                 -- Date of creation or establishment
+  Ranking number,                -- Ranking of the department
+  Budget_in_Billions number,     -- Budget of the department in billions
+  Num_Employees number          -- Number of employees in the department
+);
+
+CREATE TABLE head (
+  head_ID number,                -- Unique identifier for the head
+  name text,                     -- Name of the head
+  born_state text,               -- State where the head was born
+  age number                     -- Age of the head
+);
+
+CREATE TABLE management (
+  department_ID number,          -- Foreign key referencing Department_ID in department table
+  head_ID number,                -- Foreign key referencing head_ID in head table
+  temporary_acting text          -- Indicates if the head is temporarily acting
+);
+</schema>
 <question>What are the names of the heads who are born outside the California state?</question>
-<sql>"""
+<sql>
+"""
 ```
 
 ### response