Update

TechdocsAPI/backend/__init__.py

@@ -1,4 +1,5 @@
 import mysql.connector
+from mysql.connector import errorcode
 
 from fastapi import FastAPI, status
 from fastapi.exceptions import HTTPException
@@ -7,10 +8,10 @@ from fastapi.templating import Jinja2Templates
 from backend.utils import DBConnection
 from backend.core.ConfigEnv import config
 
-# from langchain.llms import Clarifai
+from langchain_community.llms import Clarifai
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
-from langchain_google_genai import GoogleGenerativeAI
+
 
 app = FastAPI(
     title="Techdocs",
@@ -30,12 +31,15 @@ try:
 
     prompt = PromptTemplate(template=prompt, input_variables=["instruction"])
 
-    llm = GoogleGenerativeAI(
-        model="gemini-pro",
-        google_api_key=config.GOOGLE_API_KEY,
+    llm = Clarifai(
+        pat=config.CLARIFAI_PAT,
+        user_id=config.USER_ID,
+        app_id=config.APP_ID,
+        model_id=config.MODEL_ID,
+        model_version_id=config.MODEL_VERSION_ID,
     )
 
-    llmchain = LLMChain(prompt=prompt, llm=llm)
+    llmchain = prompt | llm
     app.state.llmchain = llmchain
 
     app.state.templates = Jinja2Templates(directory="./backend/templates")

TechdocsAPI/backend/core/ConfigEnv.py

@@ -14,7 +14,12 @@ class Settings(BaseSettings):
     JWT_SECRET_KEY:str
     JWT_REFRESH_SECRET_KEY:str
     JWT_VERIFICATION_SECRET_KEY:str
-    GOOGLE_API_KEY:str
+    # GOOGLE_API_KEY:str
+    APP_ID:str
+    USER_ID:str
+    MODEL_ID:str
+    CLARIFAI_PAT:str
+    MODEL_VERSION_ID:str
 
     MAIL_SERVER_URL:str
     MAIL_USERNAME:str

TechdocsAPI/backend/services/auth/ops.py

@@ -12,6 +12,9 @@ from fastapi import HTTPException, BackgroundTasks
 from pydantic import ValidationError
 from jose import jwt
 
+# BASE_URL = "http://127.0.0.1:8000"
+BASE_URL = "https://caffeinecrew-techdocs.hf.space"
+
 async def ops_signup(bgtasks: BackgroundTasks, response_result: GeneralResponse, data: UserAuth):
     """Wrapper method to handle signup process.
 
@@ -30,7 +33,7 @@ async def ops_signup(bgtasks: BackgroundTasks, response_result: GeneralResponse,
         # user with the entered credentials already exists
         raise ExistingUserException(response_result)
     verifiction_token = Auth.create_access_token(f"{data.username} {data.email}", secret_name='VERIFICATION')
-    verification_link = f"https://caffeinecrew-techdocs.hf.space/auth/verify/{verifiction_token}"
+    verification_link = f"{BASE_URL}/auth/verify/{verifiction_token}"
 
     email_body_params = {
         "username": data.username,
@@ -109,8 +112,7 @@ def ops_regenerate_api_key(username:str) -> APIKey:
         apikey = APIKey(api_key=apikey)
     
     return apikey
-    
-        
+
 
 def ops_inference(source_code:str,api_key:str,username:str):
     response_result = GeneralResponse.get_instance(data={},
@@ -129,8 +131,9 @@ def ops_inference(source_code:str,api_key:str,username:str):
 
 
         llm_response = app.state.llmchain.invoke({"instruction": source_code_message})
+        print(llm_response)
 
-        docstring = Inference(docstr=llm_response["text"])        
+        docstring = Inference(docstr=llm_response)        
     
         return docstring
 
@@ -164,4 +167,3 @@ def ops_verify_email(request: Request, response_result: GeneralResponse, token:s
 
     except (jwt.JWTError, ValidationError):
         return app.state.templates.TemplateResponse("verification_failure.html", context={"request": request})
-    

TechdocsAPI/backend/utils/Gemini_Prompt.txt

@@ -1,3 +1,4 @@
+<|im_start|>user
 You are an expert Code Documentation writer. Your job is to document Python codebases. The Documentation should be in "Goolge Style Python Docstrings". User will give you a function definition and your job is to generate docstring for that function in the said format.
 Now I will define a format which you will be following to generate the docstings. For each function definition, you will generate the following:
 1. Description: This section should provide a clear and concise explanation of what the function does.
@@ -172,4 +173,5 @@ def token_validation(token: str) -> bool:
 """
 
 Code
-{instruction}
+{instruction} <|im_end|>
+<|im_start|>assistant

TechdocsAPI/requirements.txt

@@ -5,8 +5,11 @@ pydantic==1.10.12
 python-jose[cryptography]
 passlib[bcrypt]
 mysql-connector-python
-langchain-google-genai
+mysql
+# langchain-google-genai
 pydantic[email]
 langchain
+langchain-community
 Pillow
 jinja2
+clarifai

scripts/test.py

@@ -1,18 +1,19 @@
 from dotenv import load_dotenv
+
 load_dotenv()
 import os
 
 import mysql.connector
 from mysql.connector import errorcode
 
-config={
-    'host':os.environ.get("HOSTNAME"),
-    'user':os.environ.get("UID"),
-    'password':os.environ.get("PASSWORD"),
-    'database':os.environ.get("DATABASE")
+config = {
+    "host": os.environ.get("HOSTNAME"),
+    "user": os.environ.get("UID"),
+    "password": os.environ.get("PASSWORD"),
+    "database": os.environ.get("DATABASE"),
 }
 
-print(config)
+print(os.environ.get("HOSTNAME"))
 
 try:
     cnx = mysql.connector.connect(**config)
@@ -29,8 +30,12 @@ else:
 
     cursor.execute("DROP TABLE IF EXISTS api_key")
     cursor.execute("DROP TABLE IF EXISTS auth")
-    cursor.execute("CREATE TABLE IF NOT EXISTS auth(username VARCHAR(15) PRIMARY KEY, password TEXT, email VARCHAR(50))")
-    cursor.execute("CREATE TABLE IF NOT EXISTS api_key(username VARCHAR(15),apikey TEXT, FOREIGN KEY (username) REFERENCES auth(username))")
+    cursor.execute(
+        "CREATE TABLE IF NOT EXISTS auth(username VARCHAR(15) PRIMARY KEY, password TEXT, email VARCHAR(50))"
+    )
+    cursor.execute(
+        "CREATE TABLE IF NOT EXISTS api_key(username VARCHAR(15),apikey TEXT, FOREIGN KEY (username) REFERENCES auth(username))"
+    )
     cursor.execute("ALTER TABLE auth ADD is_verified BOOLEAN NOT NULL DEFAULT(false)")
 
     # QUERY = ('INSERT INTO {coll_name} '
@@ -46,9 +51,6 @@ else:
     # for i in cursor.fetchall():
     #     print(i)
 
-
-
-
     cnx.commit()
     cursor.close()
     cnx.close()