Initial cmd tool script

TechdocsAPI/backend/router.py

@@ -59,7 +59,5 @@ async def regenerate_api_key(access_token: str = Depends(JWTBearer())):
 @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())):
     user_sub=Auth.get_user_credentials(access_token)
-
-    print("after res")
     
     return ops_inference(code_block,api_key,user_sub)

TechdocsAPI/backend/services/auth/ops.py

@@ -102,6 +102,7 @@ def ops_inference(source_code:str,api_key:str,username:str):
         llm_response = app.state.llmchain.run({"instruction": source_code_message})
 
         docstring = Inference(docstr=llm_response)
+        print(docstring)
         
     
         return docstring

techdocs/execute.py

@@ -0,0 +1,28 @@
+from utils import parse,functools
+
+import argparse
+
+def main():
+    parser = argparse.ArgumentParser(description='Weather Application')
+    parser.add_argument('--api_key','-k',help='API key for Techdocs')
+    parser.add_argument('--username','-u',help='Username for Techdocs')
+    parser.add_argument('--password','-p',help='Password for Techdocs')
+    parser.add_argument('--dir','-d',help='Root directory to be documented')
+
+    args=parser.parse_args()
+
+    config = {
+        arg[0]:arg[1] for arg in args._get_kwargs()
+    }
+
+    data = {
+            "username":config['username'],
+            "password":config['password']
+            }
+
+    config.update({"access_token":functools.get_access_token(data)})
+
+    parse.extract_functions_from_directory(config)
+    
+if __name__ == '__main__':
+    main()    

techdocs/requirements.txt

@@ -0,0 +1 @@
+tqdm

techdocs/utils/functools.py

@@ -0,0 +1,43 @@
+import json
+import requests
+
+BASE_URL = "http://localhost:8000"
+
+
+
+def get_access_token(data, return_refresh_token=False):
+    url = BASE_URL + "/auth/login"
+    headers = {
+        "accept": "application/json",
+    }
+    data = json.dumps(data)
+    response = requests.post(url, data=data, headers=headers)
+    access_token = response.json()['access_token']
+    if return_refresh_token:
+        refresh_token = response.json()['refresh_token']
+        return access_token, refresh_token
+    return access_token
+
+
+
+
+def request_inference(config, code_block, max_retries=1):
+
+    if max_retries == 0:
+        return ""
+
+    url = BASE_URL+"/api/inference"
+    headers={"accept":"application/json", "Authorization": f"Bearer {config['access_token']}"}
+    code_input = code_block
+    response = requests.post(url=url, headers=headers, params={'code_block':code_input, 'api_key':config['api_key']})
+    if response.status_code == 200:
+        return response.json()["docstr"]
+    else:
+        data = {
+            "username":config['username'],
+            "password":config['password']
+        }
+        print("Encountered error retrying...")
+        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

@@ -0,0 +1,35 @@
+import ast
+import os
+from tqdm import tqdm
+import requests
+
+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"
+    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)
+        
+
+    for child in ast.iter_child_nodes(node):
+        if isinstance(child, ast.FunctionDef):
+            line_number = child.lineno
+        extract_outermost_function(child, line_number)
+
+# Function to traverse directories recursively and extract functions from Python files
+def extract_functions_from_directory(config):
+    for root, _, files in os.walk(config["dir"]):
+        for file in files:
+            if file.endswith(".py"):
+                file_path = os.path.join(root, file)
+                print(file_path)
+                with open(file_path, "r",errors='ignore') as file:
+                    content = file.read()
+                parsed = ast.parse(content)
+                extract_outermost_function(parsed, config)