Update src/workflow_nodes.py
Browse files- src/workflow_nodes.py +6 -4
src/workflow_nodes.py
CHANGED
|
@@ -32,14 +32,16 @@ from workflow_utils import (
|
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
-
#
|
|
|
|
|
|
|
| 36 |
|
| 37 |
-
# Setting up LLM
|
| 38 |
|
|
|
|
| 39 |
## Create a ChatOpenAI model instance using LangChain
|
| 40 |
model = ChatOpenAI(
|
| 41 |
-
openai_api_base=
|
| 42 |
-
openai_api_key=
|
| 43 |
model="gpt-4o",
|
| 44 |
streaming=False # Explicitly disabling streaming
|
| 45 |
)
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
+
# Fetch secrets from Hugging Face Spaces
|
| 36 |
+
api_key = os.environ['AZURE_OPENAI_KEY']
|
| 37 |
+
endpoint = os.environ['AZURE_OPENAI_ENDPOINT']
|
| 38 |
|
|
|
|
| 39 |
|
| 40 |
+
# Setting up LLM
|
| 41 |
## Create a ChatOpenAI model instance using LangChain
|
| 42 |
model = ChatOpenAI(
|
| 43 |
+
openai_api_base= api_key,
|
| 44 |
+
openai_api_key= endpoint,
|
| 45 |
model="gpt-4o",
|
| 46 |
streaming=False # Explicitly disabling streaming
|
| 47 |
)
|