(self, queries: List[str], filter: str = None, nResults: int = 1)
| 71 | return result["ids"][0] |
| 72 | |
| 73 | def vectorQueryDB(self, queries: List[str], filter: str = None, nResults: int = 1): |
| 74 | response = self.chromaCollection.query(query_texts=queries, n_results=nResults) if filter is None \ |
| 75 | else self.chromaCollection.query(query_texts=queries, |
| 76 | where={"type": {"$eq": filter}}, |
| 77 | n_results=nResults) |
| 78 | |
| 79 | if not response: |
| 80 | print(f"MemoryDB: No entry found.\n") |
| 81 | return "" |
| 82 | return response |
| 83 | |
| 84 | def metadataQueryDB(self, type: str = None, identifier: str = None): |
| 85 | temp = self.chromaCollection.get(where={ |
no outgoing calls
no test coverage detected