diff --git a/.github/workflows/post-commit.yaml b/.github/workflows/post-commit.yaml index 0f3ec4ce..9e982308 100644 --- a/.github/workflows/post-commit.yaml +++ b/.github/workflows/post-commit.yaml @@ -27,4 +27,4 @@ jobs: python setup.py develop - name: Run tests run: | - DEBUG=1 HEGELAI_API_KEY=1 python examples/prompttests/example.py \ No newline at end of file + DEBUG=1 python examples/prompttests/example.py \ No newline at end of file diff --git a/prompttools/experiment/openai_chat_experiment.py b/prompttools/experiment/openai_chat_experiment.py index 1450aea4..eb826650 100644 --- a/prompttools/experiment/openai_chat_experiment.py +++ b/prompttools/experiment/openai_chat_experiment.py @@ -12,7 +12,6 @@ from prompttools.mock.mock import mock_chat_completion_fn from prompttools.experiment.experiment import Experiment -from hegel.scribe import HegelScribe class OpenAIChatExperiment(Experiment): @@ -54,6 +53,7 @@ def __init__( ): self.use_scribe = use_scribe if use_scribe: + from hegel.scribe import HegelScribe self.completion_fn = HegelScribe( name=scribe_name, completion_fn=openai.ChatCompletion.create ) diff --git a/prompttools/experiment/openai_completion_experiment.py b/prompttools/experiment/openai_completion_experiment.py index 098017d2..b5d2c5ae 100644 --- a/prompttools/experiment/openai_completion_experiment.py +++ b/prompttools/experiment/openai_completion_experiment.py @@ -11,7 +11,6 @@ from prompttools.mock.mock import mock_completion_fn from prompttools.experiment.experiment import Experiment -from hegel.scribe import HegelScribe class OpenAICompletionExperiment(Experiment): @@ -61,6 +60,7 @@ def __init__( ): self.use_scribe = use_scribe if use_scribe: + from hegel.scribe import HegelScribe self.completion_fn = HegelScribe( name=scribe_name, completion_fn=openai.Completion.create )