Activate local_tracing in main + Don't override max_tokens in writer.…

…py (#156) * Activate local_tracing from env var in main * Don't override max_tokens in writer.py
Azure-Samples · Sep 3, 2024 · aec8730 · aec8730
1 parent 9e9e309
commit aec8730
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/api/agents/writer/writer.py b/src/api/agents/writer/writer.py
@@ -11,7 +11,7 @@ def write(researchContext, research, productContext, products, assignment, feedb
 
     result = prompty.execute(
         "writer.prompty",
-        parameters={"stream": True, "max_tokens": 2000},
+        parameters={"stream": True},
         inputs={
             "researchContext": researchContext,
             "research": research,

diff --git a/src/api/main.py b/src/api/main.py
@@ -15,7 +15,7 @@
 
 load_dotenv()
 LOCAL_TRACING = True if os.getenv("LOCAL_TRACING", "false").lower() == "true" else False
-tracer = init_tracing()
+tracer = init_tracing(local_tracing=LOCAL_TRACING)
 
 app = FastAPI()