pedroferreira commited on
Commit
ad3d0d5
·
1 Parent(s): 028ee7d

sets target prompting branch to properly integrate validator code

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. validator_wrapper.py +3 -3
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- git+https://github.com/opentensor/prompting.git@main
2
  aiohttp
 
1
+ git+https://github.com/opentensor/prompting.git@features/move-validator-into-prompting
2
  aiohttp
validator_wrapper.py CHANGED
@@ -1,5 +1,5 @@
1
  import bittensor as bt
2
- from neurons.validator import Validator
3
  from prompting.utils.uids import get_random_uids
4
  from prompting.protocol import PromptingSynapse
5
  from prompting.dendrite import DendriteResponseEvent
@@ -48,13 +48,13 @@ class S1ValidatorWrapper(ValidatorWrapper):
48
  bt.logging.info(f'Calling dendrite')
49
  responses = await self.validator.dendrite(
50
  axons=axons,
51
- synapse=PromptingSynapse(roles=params.roles, messages=params.request_data.messages),
52
  timeout=params.timeout,
53
  )
54
 
55
  # Encapsulate the responses in a response event (dataclass)
56
  bt.logging.info(f"Creating DendriteResponseEvent:\n {responses}")
57
- response_event = DendriteResponseEvent(responses, uids)
58
  return response_event
59
 
60
 
 
1
  import bittensor as bt
2
+ from prompting.validator import Validator
3
  from prompting.utils.uids import get_random_uids
4
  from prompting.protocol import PromptingSynapse
5
  from prompting.dendrite import DendriteResponseEvent
 
48
  bt.logging.info(f'Calling dendrite')
49
  responses = await self.validator.dendrite(
50
  axons=axons,
51
+ synapse=PromptingSynapse(roles=params.roles, messages=params.messages),
52
  timeout=params.timeout,
53
  )
54
 
55
  # Encapsulate the responses in a response event (dataclass)
56
  bt.logging.info(f"Creating DendriteResponseEvent:\n {responses}")
57
+ response_event = DendriteResponseEvent(responses, uids, params.timeout)
58
  return response_event
59
 
60