update run names (#102)

Update run names for suggest
langchain-ai · Mar 22, 2024 · 4b14415 · 4b14415
1 parent 0da6b2b
commit 4b14415
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/backend/server/api/suggest.py b/backend/server/api/suggest.py
@@ -72,7 +72,7 @@ class ExtractorDefinition(BaseModel):
 
 suggestion_chain = SUGGEST_PROMPT | model.with_structured_output(
     schema=ExtractorDefinition
-)
+).with_config({"run_name": "suggest"})
 
 UPDATE_PROMPT = ChatPromptTemplate.from_messages(
     [
@@ -94,7 +94,9 @@ class ExtractorDefinition(BaseModel):
     ]
 )
 
-UPDATE_CHAIN = UPDATE_PROMPT | model.with_structured_output(schema=ExtractorDefinition)
+UPDATE_CHAIN = (
+    UPDATE_PROMPT | model.with_structured_output(schema=ExtractorDefinition)
+).with_config({"run_name": "suggest_update"})
 
 
 # PUBLIC API
@@ -104,7 +106,6 @@ class ExtractorDefinition(BaseModel):
 async def suggest(request: SuggestExtractor) -> ExtractorDefinition:
     """Endpoint to create an extractor."""
     if len(request.jsonSchema) > 10:
-        print(f"Using update chain with {request.jsonSchema}")
         return await UPDATE_CHAIN.ainvoke(
             {"input": request.description, "json_schema": request.jsonSchema}
         )