Method
__init__
(self, project_path: str, db_name: str = '.code_tree', platform: str = 'OpenAI', api_key: str = None, embedding_model: str = "text-embedding-ada-002")
Source from the content-addressed store, hash-verified
| 10 | import pandas as pd |
| 11 | class CodeTreeMemory(Memory): |
| 12 | def __init__(self, project_path: str, db_name: str = '.code_tree', platform: str = 'OpenAI', api_key: str = None, embedding_model: str = "text-embedding-ada-002"): |
| 13 | super().__init__(project_path, db_name, platform, api_key, embedding_model) |
| 14 | self.collection_name = 'code_tree_memory' |
| 15 | self.embedder = OpenAI(api_key=api_key) |
| 16 | |
| 17 | |
| 18 | def add_code_files(self, directory: str, exclude_prefix: List[str] = ["workplace_"]): |
Callers
nothing calls this directly
Tested by
no test coverage detected