cmd tool dev (#10)

* Update init

* Update router

* Update router

* Update router

* Update router

* Update router

* Update router

* Update router.py

* Update router

* Update vercel.json

* Update vercel.json

* Update router

* Modify AST nodes inplace and write to file

TechdocsAPI/backend/router.py

@@ -17,6 +17,7 @@ app.add_middleware(
 
 @app.get("/api/response_check", tags=["Resource Server"])
 def api_response_check():
+    print("api_response_check")
     response_result = GeneralResponse.get_instance(data={},
                                       status="not_allowed",
                                       message=["Not authenticated"]
@@ -38,6 +39,7 @@ def api_response_check():
 
 @app.post("/auth/signup", summary="Creates new user account", response_model=GeneralResponse, tags=["Auth Server"])
 async def signup(response: UserAuth):
+    print("signup")
     response_result = GeneralResponse.get_instance(data={},
                                       status="not_allowed",
                                       message=["Not authenticated"]
@@ -48,16 +50,19 @@ async def signup(response: UserAuth):
 
 @app.post("/auth/login", summary="Logs in user", response_model=TokenSchema, tags=["Auth Server"])
 async def login(response:LoginCreds):
+    print("login")
     return ops_login(response)
 
 @app.put("/auth/regenerate_api_key",summary="Forget Password",response_model=APIKey,tags=["Auth Server"],dependencies=[Depends(JWTBearer())])
 async def regenerate_api_key(access_token: str = Depends(JWTBearer())):
+    print("regenerate_api_key")
     user_sub=Auth.get_user_credentials(access_token)
 
     return ops_regenerate_api_key(user_sub)
 
 @app.post("/api/inference", summary="Inference", response_model=Inference, tags=["Resource Server"], dependencies=[Depends(JWTBearer())])
 async def inference(code_block:str, api_key: str,access_token:str=Depends(JWTBearer())):
+    print("inference")
     user_sub=Auth.get_user_credentials(access_token)
     
     return ops_inference(code_block,api_key,user_sub)

techdocs/utils/functools.py

@@ -41,3 +41,8 @@ def request_inference(config, code_block, max_retries=1):
         config.update({"access_token":get_access_token(data)})
         
         return request_inference(config,config["access_token"], code_block, max_retries=max_retries-1)
+    
+
+    
+
+    

techdocs/utils/parse.py

@@ -7,20 +7,18 @@ from utils.functools import *
 
 
 # Function to extract and print the outermost nested function with line number
-def extract_outermost_function(node,  config, line_number=1,):
-    base_url = "http://localhost:8000"
+def extract_outermost_function(node, config):
+    
     if isinstance(node, ast.FunctionDef):
         function_def = ast.unparse(node)
-        print(f"Function starting at line {line_number}:\n{function_def}")
-        print("=" * 30)
-
-        response = request_inference(config=config,code_block=function_def)
+        response = request_inference(config=config, code_block=function_def)
+        docstr = response['docstr']
+        docstring = ast.Expr(value=ast.Str(s=docstr))
+        node.body.insert(0, docstring)
         
 
     for child in ast.iter_child_nodes(node):
-        if isinstance(child, ast.FunctionDef):
-            line_number = child.lineno
-        extract_outermost_function(child, line_number)
+        extract_outermost_function(child, config)
 
 # Function to traverse directories recursively and extract functions from Python files
 def extract_functions_from_directory(config):
@@ -32,4 +30,7 @@ def extract_functions_from_directory(config):
                 with open(file_path, "r",errors='ignore') as file:
                     content = file.read()
                 parsed = ast.parse(content)
-                extract_outermost_function(parsed, config)
+                extract_outermost_function(parsed, config)
+                docstr_code = ast.unparse(parsed)
+                with open(file_path, "w",errors='ignore') as file:
+                    file.write(docstr_code)

testing/test.py

@@ -0,0 +1,21 @@
+def add(a, b):
+    return a + b
+
+
+
+def multiply(a, b):
+    return a * b
+
+def subtract(a, b):
+    return a - b
+
+def divide(a, b):
+    if b == 0:
+        raise ValueError("Cannot divide by zero")
+    return a / b
+
+
+def func(*args, **kwargs):
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+    return wrapper