diff --git a/docs/README.quickstart-docker-compose.md b/docs/README.quickstart-docker-compose.md index 5e2fa237..a81da9bc 100644 --- a/docs/README.quickstart-docker-compose.md +++ b/docs/README.quickstart-docker-compose.md @@ -186,7 +186,7 @@ To change the `Ollama` model, first make sure the desired model has been pulled ### OpenAI API ``` -export OPENAI_KEY= +export OPENAI_TOKEN= docker compose -f tg-launch-openai-cassandra.yaml up -d # Using Cassandra as the graph store docker compose -f tg-launch-openai-neo4j.yaml up -d # Using Neo4j as the graph store ``` @@ -458,4 +458,4 @@ docker compose -f tg-launch--.yaml down -v > To confirm all Docker volumes have been removed, check that the following list is empty: > ``` > docker volume ls -> ``` \ No newline at end of file +> ``` diff --git a/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py b/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py index b1fe7c94..c874943e 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py @@ -23,7 +23,7 @@ default_subscriber = module default_model = 'gpt-3.5-turbo' default_temperature = 0.0 default_max_output = 4096 -default_api_key = os.getenv("OPENAI_KEY") +default_api_key = os.getenv("OPENAI_TOKEN") class Processor(ConsumerProducer):