From 38e9965d2e04fb447c029e82f6ca4e24f32ff59c Mon Sep 17 00:00:00 2001 From: Farouk Adeleke Date: Sat, 27 Dec 2025 02:18:33 -0800 Subject: [PATCH] set LM --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 4c7a59b..ffbda35 100644 --- a/main.py +++ b/main.py @@ -35,6 +35,7 @@ class GenerateCypher(PrecompiledProgram): max_tokens=config.max_tokens, ) self.generate_cypher = dspy.ChainOfThought(CypherFromText) + self.generate_cypher.set_lm(self.lm) def forward(self, text: str, neo4j_schema: list[str]): return self.generate_cypher(text=text, neo4j_schema=neo4j_schema) @@ -42,6 +43,6 @@ class GenerateCypher(PrecompiledProgram): if __name__ == "__main__": generate_cypher = GenerateCypher(GenerateCypherConfig()) - generate_cypher.push_to_hub("farouk1/text-to-cypher", with_code=True, tag="v0.0.1", commit_message="init") + generate_cypher.push_to_hub("farouk1/text-to-cypher", with_code=True, tag="v0.0.2", commit_message="set LM")