Skip to content

Commit

Permalink
🚧 fix(wip): concat into arg directly
Browse files Browse the repository at this point in the history
  • Loading branch information
mxchinegod committed Jan 20, 2024
1 parent cf5f825 commit 104c3e0
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions magnet/electrode.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,10 @@ def __init__(self, config: dict = None):
, "GENERATION_MODEL": "mistralai/Mistral-7B-Instruct-v0.1"
, "CREATE": True
}
self.nats_server = f"{self.config['NATS_USER']}:{self.config['NATS_PASSWORD']}@{self.config['NATS_URL']}"
async def auto(self):
match self.config['JOB_TYPE']:
case 'index':
self.reso = field.Resonator(self.nats_server)
self.reso = field.Resonator(f"{self.config['NATS_USER']}:{self.config['NATS_PASSWORD']}@{self.config['NATS_URL']}")
self.embedder = memory.Embedder(self.config, create=self.config["CREATE"])
await self.reso.on(category=self.config['NATS_CATEGORY'], session=self.config['NATS_SESSION'], stream=self.config['NATS_STREAM'])
await self.reso.listen(cb=self.embedder.index)
Expand Down

0 comments on commit 104c3e0

Please sign in to comment.