File size: 2,517 Bytes
c513170 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
"""
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 # Still using openai package for Groq API
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()
|