Skip to content

[Work in Progress] Add support for gemini models#67

Open
sovich-rcp wants to merge 1 commit intoSamuelSchmidgall:mainfrom
sovich-rcp:add_gemini
Open

[Work in Progress] Add support for gemini models#67
sovich-rcp wants to merge 1 commit intoSamuelSchmidgall:mainfrom
sovich-rcp:add_gemini

Conversation

@sovich-rcp
Copy link

@sovich-rcp sovich-rcp commented Jan 24, 2025

This is still a Draft Pull Request and a work in progress. The core Gemini API integration is functional in that the code is now running and connecting to Gemini without timeouts. However, it's not yet ready for review or merging.

My immediate goal was to test out gemini-2.0-flash-thinking-exp-01-21 which was just released a couple days ago, so I thought for expediency's sake I would share what I have with the project in case others are interested, even though the code is not ready for merging.

Current Status - Work Still to Be Done:

  • Cleanup Debug Statements - I've added numerous print("DEBUG: ...") statements throughout the inference.py and agents.py files to help diagnose timeout issues. These definitely need to be cleaned up before the PR is ready for review.
  • Ensure we aren't printing any API keys - Need to do a thorough sweep of these.
  • Run the code through a linter or formatter
  • Remove sleep or make configurable
  • Add support for more models.
  • Add tests

@sovich-rcp sovich-rcp changed the title Add support for gemini models [Work in Progress] Add support for gemini models Jan 24, 2025
@sovich-rcp
Copy link
Author

Need to reconcile with #64

@whats2000
Copy link

Check out the #73

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