Initial commit

.gitignore

@@ -0,0 +1,111 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+pipenv.lock
+
+# Poetry
+poetry.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# dotenv environment variables file
+.env
+.env.*  # e.g. .env.local, .env.production
+
+# virtual environments
+venv/
+ENV/
+env/
+env.bak/
+venv.bak/
+
+# VS Code settings
+.vscode/
+
+# PyCharm
+.idea/
+
+# macOS system files
+.DS_Store
+
+# Windows system files
+Thumbs.db

agent/code_index.py

@@ -0,0 +1,233 @@
+import base64
+import os
+import re
+import time
+from typing import List, Dict
+from llama_index.core import VectorStoreIndex, Document, Settings, get_response_synthesizer
+from llama_index.core.query_engine import RetrieverQueryEngine
+from llama_index.core.postprocessor import SimilarityPostprocessor
+from llama_index.embeddings.mistralai import MistralAIEmbedding
+from llama_index.llms.mistralai import MistralAI
+from mistralai import Mistral
+from agent.function_calling import github_request, get_installation_id, get_installation_token
+from config import MISTRAL_API_KEY
+
+repo_indices_cache: Dict[str, VectorStoreIndex] = {}
+INCLUDE_FILE_EXTENSIONS = {".py", ".js", ".ts", ".json", ".md", ".txt"}
+
+def fetch_repo_files(owner: str, repo: str, ref: str = "main") -> List[str]:
+    """
+    Lists all files in the repository by recursively fetching the Git tree from GitHub API.
+    Returns a list of file paths.
+    """
+    installation_id = get_installation_id(owner, repo)
+    token = get_installation_token(installation_id)
+    url = f"https://api.github.com/repos/{owner}/{repo}/git/trees/{ref}?recursive=1"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    response = github_request("GET", url, headers=headers)
+    if response.status_code != 200:
+        raise Exception(f"Failed to list repository files: {response.status_code} {response.text}")
+
+    tree = response.json().get("tree", [])
+    file_paths = [item["path"] for item in tree if item["type"] == "blob"]
+    return file_paths
+
+# print(fetch_repo_files("aditi-dsi", "EvalAI-Starters", "master"))
+
+def fetch_file_content(owner: str, repo: str, path: str, ref: str = "main") -> str:
+    """
+    Fetches the content of a file from the GitHub repository.
+    """
+    installation_id = get_installation_id(owner, repo)
+    token = get_installation_token(installation_id)
+    url = f"https://api.github.com/repos/{owner}/{repo}/contents/{path}?ref={ref}"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    response = github_request("GET", url, headers=headers)
+    if response.status_code != 200:
+        raise Exception(f"Failed to fetch file content {path}: {response.status_code} {response.text}")
+
+    content_json = response.json()
+    content = base64.b64decode(content_json["content"]).decode("utf-8", errors="ignore")
+    return content
+
+# print(fetch_file_content("aditi-dsi", "testing-cryptope", "frontend/src/lib/buildSwap.ts", "main"))
+
+def clean_line(line: str) -> str:
+    line = re.sub(r'^\s*\d+[\.\)]\s*', '', line)
+    line = line.strip(' `"\'')
+
+    return line.strip()
+
+def select_relevant_files_mistral(issue_description: str, file_paths: List[str]) -> List[str]:
+
+    model = "devstral-small-latest"
+    client = Mistral(api_key=MISTRAL_API_KEY)
+    
+    system_prompt = '''
+    You are a code reasoning assistant. Given a GitHub issue description and a list of file paths from a codebase, return a list of top 5 files that are most relevant to solving or understanding the issue, based on naming, possible associations, or inferred logic.
+
+    DO NOT RETURN ANYTHING ELSE. 
+    DO NOT RETURN ANY ADDITIONAL INFORMATION OR EXPLANATIONS. 
+    ONLY RETURN THE FILE PATHS, ONE PER LINE, WITHOUT ANY ADDITIONAL TEXT OR FORMATTING.
+    DO NOT HALLUCINATE.
+    '''
+    user_prompt = f"""Issue:
+{issue_description}
+
+Files:
+{chr(10).join(file_paths)}
+
+Return the list of most relevant files (only exact paths)."""
+
+    response = client.chat.complete(
+        model=model,
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt},
+        ],
+    )
+
+    reply = response.choices[0].message.content if hasattr(response.choices[0].message, "content") else str(response.choices[0].message)
+
+    lines = [line.strip() for line in reply.strip().splitlines()]
+    relevant_files = []
+
+    for line in lines:
+        cleaned = clean_line(line)
+        if cleaned in file_paths:
+            relevant_files.append(cleaned)
+        # else:
+        #     print(f"[Warning] Ignored unexpected line from LLM response: {line}")
+
+    if not relevant_files:
+        print("[Info] No valid file paths found in LLM response, defaulting to all files.")
+        return file_paths
+    else:
+        # print("RELEVANT files selected by LLM:")
+        return relevant_files
+
+# print(select_relevant_files_mistral('''
+# 🛠️ Configuration Error: Placeholder values detected in host_config.json
+# This file still includes default placeholders like:
+
+# <evalai_user_auth_token>
+# <host_team_pk>
+# <evalai_host_url>
+# Please replace them with real values to proceed.
+# ''',
+# ['.github/FUNDING.yml', '.github/workflows/process_challenge.yml', '.gitignore', 'README.md', 'annotations/test_annotations_devsplit.json', 'annotations/test_annotations_testsplit.json', 'challenge_config.yaml', 'challenge_data/__init__.py', 'challenge_data/challenge_1/__init__.py', 'challenge_data/challenge_1/main.py', 'evaluation_script/__init__.py', 'evaluation_script/main.py', 'github/challenge_processing_script.py', 'github/config.py', 'github/host_config.json', 'github/requirements.txt', 'github/utils.py', 'logo.jpg', 'remote_challenge_evaluation/README.md', 'remote_challenge_evaluation/eval_ai_interface.py', 'remote_challenge_evaluation/evaluate.py', 'remote_challenge_evaluation/main.py', 'remote_challenge_evaluation/requirements.txt', 'run.sh', 'submission.json', 'templates/challenge_phase_1_description.html', 'templates/challenge_phase_2_description.html', 'templates/description.html', 'templates/evaluation_details.html', 'templates/submission_guidelines.html', 'templates/terms_and_conditions.html', 'worker/__init__.py', 'worker/run.py']))
+
+def build_repo_index(owner: str, repo: str, ref: str = "main", issue_description: str = "") -> VectorStoreIndex:
+    model_name = "codestral-embed"
+    embed_model = MistralAIEmbedding(model_name=model_name, api_key=MISTRAL_API_KEY)
+    print(f"[Indexing] Starting to index repository: {owner}/{repo} at ref {ref}...")
+    file_paths = fetch_repo_files(owner, repo, ref)
+
+    if issue_description:
+        file_paths = select_relevant_files_mistral(issue_description, file_paths)
+
+    documents = []
+    for path in file_paths:
+        _, ext = os.path.splitext(path)
+        if ext.lower() not in INCLUDE_FILE_EXTENSIONS:
+            continue
+
+        try:
+            content = fetch_file_content(owner, repo, path, ref)
+            documents.append(Document(text=content, metadata={"file_path": path}))
+            print(f"[Indexing] Added file: {path}")
+            time.sleep(0.1)
+        except Exception as e:
+            print(f"[Warning] Skipping file {path} due to error: {e}")
+
+    index = VectorStoreIndex.from_documents(documents, embed_model=embed_model)
+    print(f"[Indexing] Finished indexing {len(documents)} files.")
+    return index
+
+# print(build_repo_index("aditi-dsi", "EvalAI-Starters", "master", 
+# '''
+# 🛠️ Configuration Error: Placeholder values detected in host_config.json
+# This file still includes default placeholders like:
+
+# <evalai_user_auth_token>
+# <host_team_pk>
+# <evalai_host_url>
+# Please replace them with real values to proceed.
+# '''))
+
+
+def get_repo_index(owner: str, repo: str, ref: str, issue_description: str) -> VectorStoreIndex:
+    cache_key = f"{owner}/{repo}:{hash(issue_description)}"
+    if cache_key in repo_indices_cache:
+        print(f"[Cache] Returning cached index for {cache_key}")
+        return repo_indices_cache[cache_key]
+
+    index = build_repo_index(owner, repo, ref, issue_description)
+    repo_indices_cache[cache_key] = index
+    return index
+
+
+# print(get_repo_index("aditi-dsi", "EvalAI-Starters", "master", 
+# '''
+# 🛠️ Configuration Error: Placeholder values detected in host_config.json
+# This file still includes default placeholders like:
+
+# <evalai_user_auth_token>
+# <host_team_pk>
+# <evalai_host_url>
+# Please replace them with real values to proceed.
+# '''))
+
+
+def retrieve_context(owner: str, repo: str, ref: str, issue_description: str) -> List[str]:
+    index = get_repo_index(owner, repo, ref, issue_description)
+    Settings.llm = MistralAI(model="codestral-latest", api_key=MISTRAL_API_KEY)
+    Settings.embed_model = MistralAIEmbedding(model_name="codestral-embed", api_key=MISTRAL_API_KEY)
+    retriever = index.as_retriever(similarity_top_k=5)
+    query_engine = RetrieverQueryEngine(
+    retriever=retriever,
+    response_synthesizer=get_response_synthesizer(),
+    node_postprocessors=[SimilarityPostprocessor(similarity_top_k=5)],
+    )
+    query = f"Please give relevant information from the codebase about that can help to solve or understand this issue:{issue_description}"
+    response = query_engine.query(query)
+    print(response)
+    return None
+
+# index_tools = [
+#         {
+#         "type": "function",
+#         "function": {
+#             "name": "retrieve_context",
+#             "description": "Fetch relevant context from codebase for a GitHub issue",
+#             "parameters": {
+#                 "type": "object",
+#                 "properties": {
+#                     "owner": {
+#                         "type": "string",
+#                         "description": "The owner of the repository."
+#                     },
+#                     "repo": {
+#                         "type": "string",
+#                         "description": "The name of the repository."
+#                     },
+#                     "ref": {
+#                         "type": "string",
+#                         "description": "The branch or commit reference to index from."
+#                     },
+#                     "issue_description": {
+#                         "type": "string",
+#                         "description": "The description of the issue to retrieve context for."
+#                     }
+#                 },
+#                 "required": ["owner", "repo", "ref", "issue_description"]
+#             },
+#         },
+#     },
+# ]

agent/function_calling.py

@@ -0,0 +1,210 @@
+import requests
+from urllib.parse import urlparse
+from config import APP_ID, APP_PRIVATE_KEY
+import time
+import jwt
+from datetime import datetime, timezone, timedelta
+import threading
+
+installation_tokens = {}
+token_lock = threading.Lock()
+
+def generate_jwt():
+    """Generate a JWT signed with GitHub App private key."""
+    now = int(time.time())
+    payload = {
+        "iat": now,
+        "exp": now + (10 * 60),
+        "iss": APP_ID,
+    }
+    encoded_jwt = jwt.encode(payload, APP_PRIVATE_KEY, algorithm="RS256")
+    return encoded_jwt
+
+
+def github_request(method, url, headers=None, **kwargs):
+    if headers is None:
+        jwt_token = generate_jwt()
+        headers = {
+            "Authorization": f"Bearer {jwt_token}",
+            "Accept": "application/vnd.github.v3+json",
+        }
+    while True:
+        response = requests.request(method, url, headers=headers, **kwargs)
+        
+        remaining = response.headers.get("X-RateLimit-Remaining")
+        reset_time = response.headers.get("X-RateLimit-Reset")
+
+        if remaining is None or reset_time is None:
+            return response
+        
+        remaining = int(remaining)
+        reset_time = int(reset_time)
+
+        print(f"[GitHub] Remaining: {remaining}, Reset: {reset_time}")
+
+        if response.status_code == 403 and "rate limit" in response.text.lower():
+            wait = reset_time - int(time.time()) + 5
+            print(f"Hit rate limit. Sleeping for {wait} seconds.")
+            time.sleep(max(wait, 0))
+            continue
+        if remaining <= 2:
+            wait = reset_time - int(time.time()) + 5
+            print(f"Approaching rate limit ({remaining} left). Sleeping for {wait} seconds.")
+            time.sleep(max(wait, 0))
+            continue
+
+        return response
+
+    
+def get_installation_id(owner, repo):
+    """Fetch the installation ID for the app on a repo."""
+    url = f"https://api.github.com/repos/{owner}/{repo}/installation"
+    response = github_request("GET", url)
+    if response.status_code == 200:
+        data = response.json()
+        return data["id"]
+    else:
+        raise Exception(f"Failed to get installation ID for {owner}/{repo}: {response.status_code} {response.text}")
+    
+# print(get_installation_id("aditi-dsi", "testing-cryptope"))
+
+
+def get_installation_token(installation_id):
+    """Return a valid installation token, fetch new if expired or missing."""
+    with token_lock:
+        token_info = installation_tokens.get(installation_id)
+        if token_info and token_info["expires_at"] > datetime.now(timezone.utc) + timedelta(seconds=30):
+            return token_info["token"]
+
+        url = f"https://api.github.com/app/installations/{installation_id}/access_tokens"
+        response = github_request("POST", url)
+        if response.status_code != 201:
+            raise Exception(f"Failed to fetch installation token: {response.status_code} {response.text}")
+
+        token_data = response.json()
+        token = token_data["token"]
+        expires_at = datetime.strptime(token_data["expires_at"], "%Y-%m-%dT%H:%M:%SZ").replace(tzinfo=timezone.utc)
+
+        installation_tokens[installation_id] = {"token": token, "expires_at": expires_at}
+        return token
+
+# print(get_installation_token(69452220))
+
+def fetch_github_issue(issue_url):
+    parsed = urlparse(issue_url)
+    path_parts = parsed.path.strip('/').split('/')
+    if len(path_parts) >= 4 and path_parts[2] == 'issues':
+        owner = path_parts[0]
+        repo = path_parts[1]
+        issue_num = path_parts[3]
+        return owner, repo, issue_num
+    else:
+        raise ValueError("Invalid GitHub Issue URL")
+    
+
+def get_issue_details(owner, repo, issue_num):
+    installation_id = get_installation_id(owner, repo)
+    token = get_installation_token(installation_id)
+    url = f"https://api.github.com/repos/{owner}/{repo}/issues/{issue_num}"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    response = github_request("GET", url, headers=headers)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        raise Exception(f"Failed to fetch issue: {response.status_code} {response.text}")
+
+# print(get_issue_details("aditi-dsi", "testing-cryptope", "3"))
+
+def post_comment(owner, repo, issue_num, comment_body):
+    installation_id = get_installation_id(owner, repo)
+    token = get_installation_token(installation_id)
+    url = f"https://api.github.com/repos/{owner}/{repo}/issues/{issue_num}/comments"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Accept": "application/vnd.github.v3+json"
+    }
+    data = {"body": comment_body}
+    response = github_request("POST", url, headers=headers, json=data)
+    if response.status_code == 201:
+        return response.json()
+    else:
+        raise Exception(f"Failed to post comment: {response.status_code} {response.text}")
+
+# print(post_comment("aditi-dsi", "testing-cryptope", "3", "This is a test comment from OpenSorus."))
+
+# tools = [
+#     {
+#         "type": "function",
+#         "function": {
+#             "name": "fetch_github_issue",
+#             "description": "Fetch GitHub issue details",
+#             "parameters": {
+#                 "type": "object",
+#                 "properties": {
+#                     "issue_url": {
+#                         "type": "string",
+#                         "description": "The full URL of the GitHub issue"
+#                     }
+#                 },
+#                 "required": ["issue_url"]
+#             },
+#         },
+#     },
+# {
+#         "type": "function",
+#         "function": {
+#             "name": "get_issue_details",
+#             "description": "Get details of a GitHub issue",
+#             "parameters": {
+#                 "type": "object",
+#                 "properties": {
+#                     "owner": {
+#                         "type": "string",
+#                         "description": "The owner of the repository."
+#                     },
+#                     "repo": {
+#                         "type": "string",
+#                         "description": "The name of the repository."
+#                     },
+#                     "issue_num": {
+#                         "type": "string",
+#                         "description": "The issue number."
+#                     }
+#                 },
+#                 "required": ["owner", "repo", "issue_num"],
+#             },
+#         },
+#     },
+# {
+#         "type": "function",
+#         "function": {
+#             "name": "post_comment",
+#             "description": "Post a comment on a GitHub issue",
+#             "parameters": {
+#                 "type": "object",
+#                 "properties": {
+#                     "owner": {
+#                         "type": "string",
+#                         "description": "The owner of the repository."
+#                     },
+#                     "repo": {
+#                         "type": "string",
+#                         "description": "The name of the repository."
+#                     },
+#                     "issue_num": {
+#                         "type": "string",
+#                         "description": "The issue number."
+#                     },
+#                     "comment_body": {
+#                         "type": "string",
+#                         "description": "The body of the comment."
+#                     }
+#                 },
+#                 "required": ["owner", "repo", "issue_num", "comment_body"],
+#             },
+#         },
+#     },
+# ]

agent/mistral.py

@@ -0,0 +1,198 @@
+import json
+from mistralai import Mistral
+from agent.function_calling import fetch_github_issue, get_issue_details, post_comment
+from agent.code_index import retrieve_context
+from config import MISTRAL_API_KEY
+
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "fetch_github_issue",
+            "description": "Fetch GitHub issue details",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "issue_url": {
+                        "type": "string",
+                        "description": "The full URL of the GitHub issue"
+                    }
+                },
+                "required": ["issue_url"]
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_issue_details",
+            "description": "Get details of a GitHub issue",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {
+                        "type": "string",
+                        "description": "The owner of the repository."
+                    },
+                    "repo": {
+                        "type": "string",
+                        "description": "The name of the repository."
+                    },
+                    "issue_num": {
+                        "type": "string",
+                        "description": "The issue number."
+                    }
+                },
+                "required": ["owner", "repo", "issue_num"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "retrieve_context",
+            "description": "Fetch relevant context from codebase for a GitHub issue",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {
+                        "type": "string",
+                        "description": "The owner of the repository."
+                    },
+                    "repo": {
+                        "type": "string",
+                        "description": "The name of the repository."
+                    },
+                    "ref": {
+                        "type": "string",
+                        "description": "The branch reference from either master or main to index from."
+                    },
+                    "issue_description": {
+                        "type": "string",
+                        "description": "The description of the issue to retrieve context for."
+                    }
+                },
+                "required": ["owner", "repo", "ref", "issue_description"]
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "post_comment",
+            "description": "Post a comment on a GitHub issue",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "owner": {
+                        "type": "string",
+                        "description": "The owner of the repository."
+                    },
+                    "repo": {
+                        "type": "string",
+                        "description": "The name of the repository."
+                    },
+                    "issue_num": {
+                        "type": "string",
+                        "description": "The issue number."
+                    },
+                    "comment_body": {
+                        "type": "string",
+                        "description": "The body of the comment."
+                    }
+                },
+                "required": ["owner", "repo", "issue_num", "comment_body"],
+            },
+        },
+    },
+]
+
+names_to_functions = {
+    "fetch_github_issue": fetch_github_issue,
+    "get_issue_details": get_issue_details,
+    "retrieve_context": retrieve_context,
+    "post_comment": post_comment,
+}
+
+allowed_tools = set(names_to_functions.keys())
+
+system_message = {
+    "role": "system",
+    "content": (
+        "You are a senior developer assistant bot for GitHub issues.\n\n"
+
+        "Your job is to respond to GitHub issues **professionally** and **helpfully**, but never repeat the issue description verbatim.\n\n"
+        "First, classify the issue as one of the following:\n"
+        "- Bug report\n"
+        "- Implementation question\n"
+        "- Feature request\n"
+        "- Incomplete or unclear\n\n"
+
+        "Then, based on the classification, write a clear, concise, and friendly response.\n\n"
+        "The comment should be well formatted and readable, using Markdown for code blocks and lists where appropriate.\n\n"
+        "DO NOT paste or repeat the issue description. DO NOT quote it. Respond entirely in your own words.\n"
+        "You can only use the following tools: fetch_github_issue, get_issue_details, retrieve_context, post_comment.\n"
+        "Do not attempt to use any other tools such as web_search."
+        "DO NOT HALLUCINATE OR MAKE UP TOOLS."
+    )
+}
+
+user_message = {
+    "role": "user",
+    "content": "Please suggest a fix on this issue https://github.com/aditi-dsi/testing-cryptope/issues/4."
+}
+
+messages = [system_message, user_message]
+
+api_key = MISTRAL_API_KEY
+model = "devstral-small-latest"
+client = Mistral(api_key=api_key)
+
+MAX_STEPS = 5
+tool_calls = 0
+
+while True:
+    response = client.chat.complete(
+        model=model,
+        messages=messages,
+        tools=tools,
+        tool_choice="any",
+    )
+    msg = response.choices[0].message
+    messages.append(msg)
+
+    if hasattr(msg, "tool_calls") and msg.tool_calls:
+        for tool_call in msg.tool_calls:
+            function_name = tool_call.function.name
+            function_params = json.loads(tool_call.function.arguments)
+            if function_name in allowed_tools:
+                function_result = names_to_functions[function_name](**function_params)
+                print(f"Agent is calling tool: {function_name}")
+                tool_calls += 1
+                messages.append({
+                    "role": "tool",
+                    "tool_call_id": tool_call.id,
+                    "content": str(function_result)
+                })
+
+                if function_name == "post_comment":
+                    print("OpenSorus (final): ✅ Comment posted successfully. No further action needed.")
+                    exit(0)
+
+            else:
+                print(f"LLM tried to call unknown tool: {function_name}")
+                tool_error_msg = (
+                    f"Error: Tool '{function_name}' is not available. "
+                    "You can only use the following tools: fetch_github_issue, get_issue_details, post_comment."
+                )
+                messages.append({
+                    "role": "tool",
+                    "tool_call_id": tool_call.id,
+                    "content": tool_error_msg
+                })
+        if tool_calls >= MAX_STEPS:
+            print(f"Agent stopped after {MAX_STEPS} tool calls to protect against rate limiting.")
+            break
+    else:
+        print("OpenSorus (final):", msg.content)
+        break

config.py

@@ -0,0 +1,8 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY")
+APP_ID = os.getenv("APP_ID")
+APP_PRIVATE_KEY = os.getenv("APP_PRIVATE_KEY", "").encode().decode("unicode_escape").strip()