graphiti icon indicating copy to clipboard operation
graphiti copied to clipboard

small model fix

Open prasmussen15 opened this issue 8 months ago • 0 comments

[!IMPORTANT] Enhance LLM client configurations by introducing model size handling and updating related functions to support these changes.

  • LLM Client Enhancements:
    • Introduced ModelSize enum in config.py to specify model sizes (small, medium).
    • Updated generate_response and _generate_response methods in anthropic_client.py, client.py, gemini_client.py, groq_client.py, openai_client.py, openai_generic_client.py to accept model_size parameter.
    • Added small_model attribute to LLMConfig in config.py for handling simpler prompts.
  • Functionality Updates:
    • Modified dedupe_extracted_edge() in edge_operations.py and resolve_extracted_node() in node_operations.py to use ModelSize.small.
    • Updated get_edge_contradictions() in temporal_operations.py to use ModelSize.small.
  • Miscellaneous:
    • Fixed typo in podcast_transcript.txt.
    • Incremented version in pyproject.toml to 0.11.4.

This description was created by Ellipsis for b5e7dd9df1db5cf0e26e2695affe97f2589593ce. You can customize this summary. It will automatically update as commits are pushed.

prasmussen15 avatar May 02 '25 13:05 prasmussen15