|
|
"""
|
|
|
Simple test script to verify the LLM Code Deployment setup
|
|
|
"""
|
|
|
|
|
|
import os
|
|
|
import sys
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
|
def test_environment_variables():
|
|
|
"""Test if all required environment variables are set"""
|
|
|
load_dotenv()
|
|
|
|
|
|
required_vars = [
|
|
|
"OPENAI_API_KEY",
|
|
|
"GITHUB_TOKEN",
|
|
|
"GITHUB_USERNAME",
|
|
|
"SHARED_SECRET"
|
|
|
]
|
|
|
|
|
|
missing_vars = []
|
|
|
for var in required_vars:
|
|
|
if not os.getenv(var):
|
|
|
missing_vars.append(var)
|
|
|
|
|
|
if missing_vars:
|
|
|
print(f"Missing environment variables: {', '.join(missing_vars)}")
|
|
|
print("Please set these in your .env file")
|
|
|
return False
|
|
|
else:
|
|
|
print("All required environment variables are set")
|
|
|
return True
|
|
|
|
|
|
def test_imports():
|
|
|
"""Test if all required modules can be imported"""
|
|
|
try:
|
|
|
import fastapi
|
|
|
import uvicorn
|
|
|
import openai
|
|
|
import github
|
|
|
import httpx
|
|
|
import pydantic
|
|
|
print("All required packages are installed")
|
|
|
return True
|
|
|
except ImportError as e:
|
|
|
print(f"Missing package: {e}")
|
|
|
print("Please run: pip install -r requirements.txt")
|
|
|
return False
|
|
|
|
|
|
def test_module_imports():
|
|
|
"""Test if our custom modules can be imported"""
|
|
|
try:
|
|
|
from llm_helper import LLMHelper
|
|
|
from github_helper import GitHubHelper
|
|
|
from deploy_helper import DeployHelper
|
|
|
print("All custom modules can be imported")
|
|
|
return True
|
|
|
except ImportError as e:
|
|
|
print(f"Error importing custom modules: {e}")
|
|
|
return False
|
|
|
|
|
|
def main():
|
|
|
"""Run all tests"""
|
|
|
print("Testing LLM Code Deployment Setup")
|
|
|
print("=" * 40)
|
|
|
|
|
|
tests = [
|
|
|
test_imports,
|
|
|
test_environment_variables,
|
|
|
test_module_imports
|
|
|
]
|
|
|
|
|
|
passed = 0
|
|
|
total = len(tests)
|
|
|
|
|
|
for test in tests:
|
|
|
if test():
|
|
|
passed += 1
|
|
|
print()
|
|
|
|
|
|
print("=" * 40)
|
|
|
if passed == total:
|
|
|
print(f"All tests passed! ({passed}/{total})")
|
|
|
print("Your LLM Code Deployment system is ready to use!")
|
|
|
print("\nTo start the server, run:")
|
|
|
print("python main.py")
|
|
|
else:
|
|
|
print(f"{passed}/{total} tests passed")
|
|
|
print("Please fix the issues above before running the server")
|
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
main()
|
|
|
|
