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

Adding Litellm client and some models/providers. #563

Open
wants to merge 6 commits into
base: main
Choose a base branch
from

Conversation

OSH212
Copy link

@OSH212 OSH212 commented May 15, 2024

Description

Created litellm_client.py for LiteLLM and updated llm.py, sample.config.toml and README.md accordingly.
I tested it with OpenRouter and DeepInfra hosted models.

Explain what existing problem does the pull request solve?

Allowing usage of LiteLLM framework

Test plan (required)

Demonstrate the code is solid. Example: The exact commands you ran and their output, screenshots / videos if the pull request changes UI.

24.05.15 17:05:02: root: INFO : User message: {'message': 'create a modern snake game', 'base_model': 'OpenRouter: Gemma 7B', 'project_name': 'gemma', 'search_engine': 'Google'}
24.05.15 17:05:02: root: INFO : SOCKET server-message MESSAGE: {'messages': {'from_devika': False, 'message': 'create a modern snake game', 'timestamp': '2024-05-15 17:05:02'}}
24.05.15 17:05:02: root: INFO : SOCKET agent-state MESSAGE: [{'internal_monologue': "I'm starting the work...", 'browser_session': {'url': None, 'screenshot': None}, 'terminal_session': {'command': None, 'output': None, 'title': None}, 'step': 1, 'message': None, 'completed': False, 'agent_is_active': True, 'token_usage': 0, 'timestamp': '2024-05-15 17:05:02'}]
24.05.15 17:05:02: root: INFO : SOCKET tokens MESSAGE: {'token_usage': 734}

Closing issues (optional)

Fixes #

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.

None yet

1 participant