-
Notifications
You must be signed in to change notification settings - Fork 1
Classification test #9
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
Conversation
…into classification_test
…into classification_test
| response: EmbeddingResponse = await aembedding( | ||
| model=embed_model, input=inputs, dimensions=768 | ||
| ) |
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.
Try removing this dimensions parameter if you are having issues with forcing the BERT model's dimensions. I think it would be cool to get this working so give it another shot, if you run into more error send me a message.
| classification_names: list[str] = await create_classifications_helper( | ||
| classification_service, | ||
| tenant_id, | ||
| [classification.name for classification in initial_classifications], |
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.
This was my bad before. I sent tenant_id to the method instead of extracted_files. You shouldn't need to change the signature of the method, just correct what is being sent through it.
| async def create_classifications( | ||
| extracted_files: list[ExtractedFile], | ||
| classification_service: ClassificationService, | ||
| tenant_id: UUID, | ||
| initialClassifications: list[str], | ||
| ) -> list[str]: |
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.
Revert this change, if it properly sends the extracted files it should work properly
…ing and file mapping. Removed redundant second LLM call.
…tionary. Uncommented initialClassificatoins parameter which merges initial classifications with new ones.
Finished create_classifications.py and classify_files.py. Files connected and worth with frontend.