From b716af8e199faf29e8a16ee3cf732d9a8f40c225 Mon Sep 17 00:00:00 2001 From: Jason Liu Date: Wed, 17 Apr 2024 17:34:59 -0600 Subject: [PATCH] bump --- tests/util.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/util.py b/tests/util.py index 9fb8528..5d8d75a 100644 --- a/tests/util.py +++ b/tests/util.py @@ -2,6 +2,7 @@ from anthropic import AsyncAnthropic import instructor from enum import Enum +import os class Models(str, Enum): @@ -14,25 +15,25 @@ class Models(str, Enum): clients = ( instructor.from_openai( - AsyncOpenAI(), + AsyncOpenAI(api_key=os.environ.get("OPENAI_API_KEY")), model=Models.GPT35TURBO, ), instructor.from_openai( - AsyncOpenAI(), + AsyncOpenAI(api_key=os.environ.get("OPENAI_API_KEY")), model=Models.GPT4TURBO, ), instructor.from_anthropic( - AsyncAnthropic(), + AsyncAnthropic(api_key=os.environ.get("ANTHROPIC_API_KEY")), model=Models.CLAUDE3_OPUS, max_tokens=4000, ), instructor.from_anthropic( - AsyncAnthropic(), + AsyncAnthropic(api_key=os.environ.get("ANTHROPIC_API_KEY")), model=Models.CLAUDE3_SONNET, max_tokens=4000, ), instructor.from_anthropic( - AsyncAnthropic(), + AsyncAnthropic(api_key=os.environ.get("ANTHROPIC_API_KEY")), model=Models.CLAUDE3_HAIKU, max_tokens=4000, ),