Create services/client.py

services/client.py

@@ -0,0 +1,99 @@
+from concurrent.futures import ThreadPoolExecutor
+from multiprocessing.pool import ThreadPool
+from pydoc import cli
+from openai import OpenAI
+import os
+from dotenv import load_dotenv
+from typing import List
+
+load_dotenv()
+
+# constants
+BASE_URL = "https://api.novita.ai/v3/openai"
+MODEL_NAME = "meta-llama/llama-3.2-1b-instruct"
+
+client = OpenAI(
+    base_url= BASE_URL,
+    api_key=os.environ.get("NOVITA_API_KEY"),
+)
+
+def generate_commented_code(code:str, max_token:int):
+    try:
+        chat_responses = client.chat.completions.create(
+        model= MODEL_NAME,
+        temperature=0.7,
+        stream= False,
+        max_tokens= max_token,
+        messages = [
+                {
+                    "role": "system",
+                    "content": (
+                        "You are a senior software engineer skilled in writing precise comments only. Follow these instructions strictly:\n\n"
+                        "Identify the programming language used and then do following:\n"
+                        "Add **clear, concise, and professional comments** to the provided code. Include:\n"
+                        "Docstrings for all functions and classes\n"
+                        "Inline comments explaining non-obvious or critical logic\n\n"
+                        "Strictly do not do anything else than writing comments. Stricltly do not add or provide alternative code"
+
+                        "Example response:"
+                        '''
+                        def add(a, b):
+                            """
+                            Returns the sum of two numbers.
+
+                            Args:
+                                a (int): The first number.
+                                b (int): The second number.
+
+                            Returns:
+                                int: The sum of a and b.
+
+                            Raises:
+                                TypeError: If either a or b is not an integer.
+                                ValueError: If either a or b is negative.
+                            """
+                            # Check if both inputs are integers
+                            if not isinstance(a, int) or not isinstance(b, int):
+                                raise TypeError("Both inputs must be integers.")
+                            # Check if both inputs are non-negative
+                            if a < 0 or b < 0:
+                                raise ValueError("Both inputs must be non-negative.")
+
+                            # Use the built-in addition operator to perform the calculation
+                            # This is more efficient and readable than using a loop or recursive function
+                            return a + b
+                        '''
+                    )
+                },
+                {"role": "user", "content": code}
+            ]
+        )
+        
+        result = chat_responses.choices[0].message.content
+        return result
+    except Exception as ex:
+        return f"Error generating comments {ex}"
+
+def generate_documentation(code:str, max_token:int):
+    try:
+        chat_responses = client.chat.completions.create(
+            model= MODEL_NAME,
+            temperature=0.7,
+            stream= False,
+            max_tokens= max_token,
+            messages = [
+                    {
+                        "role": "system",
+                        "content": (
+                            "You are a senior software engineer skilled in writing structured documentation only. Follow these instructions strictly:\n\n"
+                            "Identify the programming language used and then do following:\n"
+                            "Produce a **strictly formatted JSON** object with:\n"
+                            "name: Function or class name\n"
+                            "description: Brief summary of what it does\n"
+                            "parameters: Dictionary with param name, type, and description\n"
+                            "returns: Return name, type, and description\n"
+                            "exceptions (optional): Raised exceptions with reasons\n"
+                            "examples: Code usage examples with expected output\n\n"
+
+                            "Example response: "
+                            """