-
Notifications
You must be signed in to change notification settings - Fork 0
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
image analysis abstract class #3
base: main
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
As discussed, please change the methods to generate prompts by returning only a string and, add it to the base class. Then you can use this in the Gemini and OpenAI subclasses.
|
||
def _load_model(self, model_name: str, model_type: Literal["vqa", "ic"]): | ||
def _load_model(self, model_name: str, model_type: Literal["vqa", "ic"]) -> ModelConfig: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please update the method name to _build_model_config
:-)
|
||
class ModelConfig(TypedDict): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Very nice idea to use a typed dict as a return type :-)
…turn hints. added the ModelConfig structure.
8a9c3be
to
b23fa35
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
With this commit you pushed a lot of changes which are either only for your local config or are not semantically coherent with the pull request (which is about the image analysis assistant only) -- please remove this commit (if you don't know how to do this, try to find out, there's a ton of Git Tutorials ;-) )
98421f0
to
b23fa35
Compare
It is done. |
added the image analysis abstract class and open ai image analysis class. ALso, modified gemini image analysis assistant.