Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Gemini LLM and Summarizer #1176

Merged
merged 10 commits into from
Feb 17, 2025
Merged

Add Gemini LLM and Summarizer #1176

merged 10 commits into from
Feb 17, 2025

Conversation

karanataryn
Copy link
Contributor

@karanataryn karanataryn commented Feb 12, 2025

This pull request introduces support for the Google Gemini language model in the Sycamore project. The changes include adding the necessary dependencies, integrating the Gemini model, adding it as a summarizer, and implementing unit tests to ensure its functionality.

@karanataryn karanataryn changed the title Add Gemini Add Gemini LLM and Summarization Feb 12, 2025
@karanataryn karanataryn changed the title Add Gemini LLM and Summarization Add Gemini LLM and Summarizer Feb 14, 2025
@karanataryn karanataryn marked this pull request as ready for review February 14, 2025 01:55
@karanataryn karanataryn requested a review from HenryL27 February 14, 2025 01:55
@@ -81,6 +82,23 @@ def __init__(self, llm: LLM, prompt: Optional[str] = None, include_context: bool
self.prompt = prompt
self.include_context = include_context

def summarize_image(self, image: Image.Image, context: Optional[str]) -> str:
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

what do you think about "context" -> "system" ? idk.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I like context for now, we can change later based on usage.

@karanataryn karanataryn merged commit 21c3fc0 into main Feb 17, 2025
12 of 15 checks passed
@karanataryn karanataryn deleted the ksampath/update-summarize branch February 17, 2025 17:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants