phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) test - moved to circe ci dockerfile

Browse source
This commit is contained in:
ishaan-jaff 2024-01-10 21:54:00 +05:30
parent 954d1b071c
commit 1ff9785c6b
1 changed files with 52 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

104
litellm/tests/test_proxy_server_keys.py
View file

@ -1,73 +1,73 @@
import sys, os, time
import traceback
from dotenv import load_dotenv
# import sys, os, time
# import traceback
# from dotenv import load_dotenv
load_dotenv()
import os, io
# load_dotenv()
# import os, io
# this file is to test litellm/proxy
# # this file is to test litellm/proxy
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
import pytest, logging
import litellm
from litellm import embedding, completion, completion_cost, Timeout
from litellm import RateLimitError
# sys.path.insert(
# 0, os.path.abspath("../..")
# ) # Adds the parent directory to the system path
# import pytest, logging
# import litellm
# from litellm import embedding, completion, completion_cost, Timeout
# from litellm import RateLimitError
import sys, os, time
import traceback
from dotenv import load_dotenv
# import sys, os, time
# import traceback
# from dotenv import load_dotenv
load_dotenv()
import os, io
# load_dotenv()
# import os, io
# this file is to test litellm/proxy
from concurrent.futures import ThreadPoolExecutor
# # this file is to test litellm/proxy
# from concurrent.futures import ThreadPoolExecutor
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
# sys.path.insert(
# 0, os.path.abspath("../..")
# ) # Adds the parent directory to the system path
import pytest, logging, requests
import litellm
from litellm import embedding, completion, completion_cost, Timeout
from litellm import RateLimitError
from github import Github
import subprocess
# import pytest, logging, requests
# import litellm
# from litellm import embedding, completion, completion_cost, Timeout
# from litellm import RateLimitError
# from github import Github
# import subprocess
# Function to execute a command and return the output
def run_command(command):
process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
output, _ = process.communicate()
return output.decode().strip()
# # Function to execute a command and return the output
# def run_command(command):
# process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
# output, _ = process.communicate()
# return output.decode().strip()
# Retrieve the current branch name
branch_name = run_command("git rev-parse --abbrev-ref HEAD")
# # Retrieve the current branch name
# branch_name = run_command("git rev-parse --abbrev-ref HEAD")
# GitHub personal access token (with repo scope) or use username and password
access_token = os.getenv("GITHUB_ACCESS_TOKEN")
# Instantiate the PyGithub library's Github object
g = Github(access_token)
# # GitHub personal access token (with repo scope) or use username and password
# access_token = os.getenv("GITHUB_ACCESS_TOKEN")
# # Instantiate the PyGithub library's Github object
# g = Github(access_token)
# Provide the owner and name of the repository where the pull request is located
repository_owner = "BerriAI"
repository_name = "litellm"
# # Provide the owner and name of the repository where the pull request is located
# repository_owner = "BerriAI"
# repository_name = "litellm"
# Get the repository object
repo = g.get_repo(f"{repository_owner}/{repository_name}")
# # Get the repository object
# repo = g.get_repo(f"{repository_owner}/{repository_name}")
# Iterate through the pull requests to find the one related to your branch
for pr in repo.get_pulls():
print(f"in here! {pr.head.ref}")
if pr.head.ref == branch_name:
pr_number = pr.number
break
# # Iterate through the pull requests to find the one related to your branch
# for pr in repo.get_pulls():
# print(f"in here! {pr.head.ref}")
# if pr.head.ref == branch_name:
# pr_number = pr.number
# break
print(f"The pull request number for branch {branch_name} is: {pr_number}")
# print(f"The pull request number for branch {branch_name} is: {pr_number}")
# def test_add_new_key():