6997607817

Committed 26 Nov 2023 07:37PM UTC coverage: 82.927% (-0.6%) from 83.504%

Build # 6997607817

Build Type

Pull #225

github

Committed by

web-flow

Commit Message

Merge fc7ee39db into ef70801c1

Pull Request Pull Request #225: Add braintrust proxy

Run Details

6 of 14 new or added lines in 1 file covered. (42.86%)

4 existing lines in 1 file now uncovered.

578 of 697 relevant lines covered (82.93%)

0.83 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

60.0

/tests/openai/evals/conftest.py

# conftest.py
from openai import AsyncOpenAI, OpenAI
import pytest
import os

try:
    import braintrust

    wrap_openai = braintrust.wrap_openai
except ImportError:

    def wrap_openai(x):
        return x

    pass


@pytest.fixture(scope="session")
def client():
    if "BRAINTRUST_API_KEY" in os.environ:
        yield wrap_openai(
            OpenAI(
                api_key=os.environ["BRAINTRUST_API_KEY"],
                base_url="https://proxy.braintrustapi.com/v1",
            )
        )
    else:
        yield OpenAI()


@pytest.fixture(scope="session")
def aclient():
    if "BRAINTRUST_API_KEY" in os.environ:
        yield wrap_openai(
            AsyncOpenAI(
                api_key=os.environ["BRAINTRUST_API_KEY"],
                base_url="https://proxy.braintrustapi.com/v1",
            )
        )
    else:
        yield AsyncOpenAI()

1	# conftest.py
2	from openai import AsyncOpenAI, OpenAI	1✔
3	import pytest	1✔
4	import os	1✔
5
6	try:	1✔
7	import braintrust	1✔
8
NEW 9	wrap_openai = braintrust.wrap_openai	×
10	except ImportError:	1✔
11
12	def wrap_openai(x):	1✔
NEW 13	return x	×
14
15	pass	1✔
16
17
18	@pytest.fixture(scope="session")	1✔
19	def client():	1✔
NEW 20	if "BRAINTRUST_API_KEY" in os.environ:	×
21	yield wrap_openai(
22	OpenAI(
23	api_key=os.environ["BRAINTRUST_API_KEY"],
24	base_url="https://proxy.braintrustapi.com/v1",
25	)
26	)
NEW UNCOV 27	else:	×
NEW UNCOV 28	yield OpenAI()	×
29
30
31	@pytest.fixture(scope="session")	1✔
32	def aclient():	1✔
NEW 33	if "BRAINTRUST_API_KEY" in os.environ:	×
34	yield wrap_openai(
35	AsyncOpenAI(
36	api_key=os.environ["BRAINTRUST_API_KEY"],
37	base_url="https://proxy.braintrustapi.com/v1",
38	)
39	)
NEW UNCOV 40	else:	×
NEW UNCOV 41	yield AsyncOpenAI()	×

jxnl / instructor / 6997607817

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous