Skip to content

Commit fa29529

Browse files
committed
Update score_chain.jinja2 to log run info by default.
1 parent d41cfe9 commit fa29529

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ads/llm/templates/score_chain.jinja2

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,15 @@ from functools import lru_cache
66
from langchain.chains import LLMChain
77
from ads import set_auth
88
from ads.llm.deploy import ChainDeployment
9+
from ads.llm.chain import LOG_ADS_GUARDRAIL_INFO
910
from ads.config import OCI_RESOURCE_PRINCIPAL_VERSION
1011

1112
if OCI_RESOURCE_PRINCIPAL_VERSION:
1213
set_auth("resource_principal")
1314

15+
if LOG_ADS_GUARDRAIL_INFO not in os.environ:
16+
os.environ[LOG_ADS_GUARDRAIL_INFO] = "1"
17+
1418

1519
"""
1620
Inference script. This script is used for prediction by scoring server when schema is known.

0 commit comments

Comments
 (0)