diff --git a/docs/examples/japanese_extraction.md b/docs/examples/japanese_extraction.md index fd4d26e6..a92a9c3a 100644 --- a/docs/examples/japanese_extraction.md +++ b/docs/examples/japanese_extraction.md @@ -51,7 +51,7 @@ for entity in result.extractions: if entity.char_interval: start, end = entity.char_interval.start_pos, entity.char_interval.end_pos position_info = f" (pos: {start}-{end})" - + print(f"• {entity.extraction_class}: {entity.extraction_text}{position_info}") # Expected Output: diff --git a/langextract/providers/gemini.py b/langextract/providers/gemini.py index 24f74695..d4bdb33a 100644 --- a/langextract/providers/gemini.py +++ b/langextract/providers/gemini.py @@ -399,7 +399,13 @@ def infer( if key in merged_kwargs: config[key] = merged_kwargs[key] - handled_keys = {'temperature', 'max_output_tokens', 'top_p', 'top_k'} + handled_keys = { + 'temperature', + 'max_output_tokens', + 'top_p', + 'top_k', + 'thinking_config', + } for key, value in merged_kwargs.items(): if ( key not in handled_keys