Confusion between OPENAI_KEY and OPENAI_TOKEN: settled on OPENAI_TOKE…

…N. (#161)
trustgraph-ai · Nov 15, 2024 · 0253281 · 0253281
1 parent 081dd69
commit 0253281
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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=<TOKEN-GOES-HERE>
+export OPENAI_TOKEN=<TOKEN-GOES-HERE>
 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-<model-deployment>-<graph-store>.yaml down -v
 > To confirm all Docker volumes have been removed, check that the following list is empty:
 > ```
 > docker volume ls
-> ```
+> ```
diff --git 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_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):