Skip to content

Commit d437133

Browse files
committed
address comments
1 parent bc9c3be commit d437133

File tree

3 files changed

+0
-8
lines changed

3 files changed

+0
-8
lines changed

stagehand/handlers/extract_handler.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,8 +96,6 @@ async def extract(
9696
else:
9797
transformed_schema = DefaultExtractSchema
9898

99-
self.logger.info(f"Transformed schema: {transformed_schema}")
100-
10199
# Use inference to call the LLM
102100
extraction_response = extract_inference(
103101
instruction=instruction,

stagehand/llm/client.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,6 @@
1111
logger = logging.getLogger(__name__)
1212

1313

14-
litellm.enable_json_schema_validation = True
15-
16-
1714
class LLMClient:
1815
"""
1916
Client for making LLM calls using the litellm library.

stagehand/llm/inference.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -195,9 +195,6 @@ def extract(
195195
extract_content = extract_response.choices[0].message.content
196196
if isinstance(extract_content, str):
197197
try:
198-
logger.info(
199-
f"Extraction response: {extract_content} for schema: {schema}"
200-
)
201198
extracted_data = json.loads(extract_content)
202199
except json.JSONDecodeError:
203200
logger.error(

0 commit comments

Comments
 (0)